File size: 702 Bytes
232e96a
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
import ai_gradio
from utils_ai_gradio import get_app

# Get the Groq models but keep their full names for loading
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
]

# Create and launch the interface using get_app utility
demo = get_app(
    models=GROQ_MODELS_FULL,  # Use the full names with prefix
    default_model=GROQ_MODELS_FULL[-2],
    dropdown_label="Select Groq Model",
    choices=GROQ_MODELS_DISPLAY,  # Display names without prefix
    fill_height=True,
    coder=True
)


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