File size: 562 Bytes
986d5a0 20d974d 986d5a0 20d974d 986d5a0 20d974d 986d5a0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import ai_gradio
from utils_ai_gradio import get_app
SMOLAGENTS_MODELS_FULL = [k for k in ai_gradio.registry.keys() if k.startswith("smolagents:")]
SMOLAGENTS_MODELS_DISPLAY = [k.replace("smolagents:", "") for k in SMOLAGENTS_MODELS_FULL]
demo = get_app(
models=SMOLAGENTS_MODELS_FULL, # Use the full names with prefix
default_model=SMOLAGENTS_MODELS_FULL[-1],
dropdown_label="Select SmolAgents Model",
choices=SMOLAGENTS_MODELS_DISPLAY, # Display names without prefix
fill_height=True,
)
if __name__ == "__main__":
demo.launch()
|