File size: 581 Bytes
e8fb5c0 232e96a 0a632f8 232e96a 0a632f8 232e96a 0a632f8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 |
import os
import ai_gradio
from utils_ai_gradio import get_app
# Get the Groq models from the registry
GROQ_MODELS_FULL = [
k for k in ai_gradio.registry.keys()
if k.startswith('groq:')
]
# Create display names without the prefix
GROQ_MODELS_DISPLAY = [
k.replace('groq:', '')
for k in GROQ_MODELS_FULL
]
demo = get_app(
models=GROQ_MODELS_FULL,
default_model=GROQ_MODELS_FULL[-2],
src=ai_gradio.registry,
dropdown_label="Select Groq Model",
choices=GROQ_MODELS_DISPLAY,
fill_height=True
)
if __name__ == "__main__":
demo.launch()
|