First Commit

app.py

@@ -56,10 +56,8 @@ async def startup_event():
 
 
 
-async def chat_function(user_message, history, model_id):
+def chat_function(user_message, history, model_id):
     global agent_instance
-    if agent_instance is None:
-        agent_instance = await get_agent()
     prompt=f"""You're an assistant helping with hugging face model cards.
         First, run the tool `read_model_card` on repo_id `{model_id}` to get the model card.
         Then answer this user question based on the model card:
@@ -67,15 +65,10 @@ async def chat_function(user_message, history, model_id):
     history = history + [(user_message, None)]
     try:
         response = ""
-        try:
-            async for output in agent_instance.run(prompt):
-                if hasattr(output, "content") and output.content:
-                    response = output.content
-        except TypeError:
-            for output in agent_instance.run(prompt):
-                if hasattr(output, "content") and output.content:
-                    response = output.content
-            
+        for output in agent_instance.run(prompt):
+            if hasattr(output, "content") and output.content:
+                response = output.content
+
         final_response = response or "⚠️ Sorry, I couldn't generate a response."
         history[-1] = (user_message, final_response)
     except Exception as e:
@@ -85,6 +78,7 @@ async def chat_function(user_message, history, model_id):
 
 
 
+
 def create_gradio_app():
     with gr.Blocks(title="Model Card Chatbot") as demo:
         gr.Markdown("## 🤖 Model Card Chatbot\nAsk questions about Hugging Face model card")

requirements.txt

@@ -4,4 +4,5 @@ gradio==5.38.0
 gradio_client==1.11.0
 huggingface-hub==0.33.4
 uvicorn==0.35.0
+python-dotenv