Spaces:
Running on CPU Upgrade

File size: 729 Bytes
2eb382e
5568b1a
2eb382e
7360ef0
2eb382e
5568b1a
7360ef0
2eb382e
5568b1a
fbaae9e
2eb382e
d1eb1fb
2eb382e
 
 
 
 
d1eb1fb
2eb382e
d1eb1fb
35d8419
5568b1a
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
import ai_gradio

from utils_ai_gradio import get_app

# Get the Gemini models but keep their full names for loading
GEMINI_MODELS_FULL = [k for k in ai_gradio.registry if k.startswith("gemini:")]

# Create display names without the prefix
GEMINI_MODELS_DISPLAY = [k.replace("gemini:", "") for k in GEMINI_MODELS_FULL]

# Create and launch the interface using get_app utility
demo = get_app(
    models=GEMINI_MODELS_FULL,  # Use the full names with prefix
    default_model=GEMINI_MODELS_FULL[-2],
    dropdown_label="Select Gemini Model",
    choices=GEMINI_MODELS_DISPLAY,  # Display names without prefix
    src=ai_gradio.registry,
    enable_voice=True,
    fill_height=True,
)

if __name__ == "__main__":
    demo.launch()