Spaces:
Sleeping
Sleeping
import os | |
os.environ["HF_TOKEN"] = os.getenv("HF_TOKEN") | |
import gradio as gr | |
from smolagents import CodeAgent, HfApiModel | |
from retriever import guest_info_tool | |
from tools import search_tool, weather_info_tool, hub_stats_tool | |
# Setup agent | |
model = HfApiModel() | |
alfred = CodeAgent( | |
tools=[guest_info_tool, search_tool, weather_info_tool, hub_stats_tool], | |
model=model | |
) | |
# Gradio interface function | |
def ask_alfred(question): | |
return alfred.run(question) | |
# Gradio app | |
demo = gr.Interface(fn=ask_alfred, inputs="text", outputs="text", title="π© Alfred, Gala Assistant") | |
demo.launch() | |