anychat / app_smolagents.py
akhaliq's picture
akhaliq HF staff
add smolagents example
986d5a0
raw
history blame contribute delete
583 Bytes
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()