File size: 710 Bytes
c44529f e8fb5c0 c44529f 0a632f8 c44529f 0a632f8 c44529f 0a632f8 c44529f 531b2a0 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 |
import ai_gradio
from utils_ai_gradio import get_app
# Get the mistral models but keep their full names for loading
MISTRAL_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("mistral:")]
# Create display names without the prefix
MISTRAL_MODELS_DISPLAY = [k.replace("mistral:", "") for k in MISTRAL_MODELS_FULL]
# Create and launch the interface using get_app utility
demo = get_app(
models=MISTRAL_MODELS_FULL, # Use the full names with prefix
default_model=MISTRAL_MODELS_FULL[5],
dropdown_label="Select Mistral Model",
choices=MISTRAL_MODELS_DISPLAY, # Display names without prefix
fill_height=True,
coder=True
)
if __name__ == "__main__":
demo.launch()
|