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