Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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.keys() | |
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, | |
) | |