First Commit

app.py

@@ -12,7 +12,7 @@ load_dotenv()
 HF_TOKEN=os.getenv("HF_TOKEN")
 HF_MODEL=os.getenv("HF_MODEL","google/gemma-2-2b")
 app=FastAPI(title="MODEL-CARD-CHATBOT")
-app.add_middleware(CORSMiddleware,allow_origins=["*"])
+app.add_middleware(CORSMiddleware, allow_origins=["*"], allow_methods=['*'], allow_headers=['*'])
 
 agent_instance: Optional[Agent]=None
 DEFAULT_PROVIDER:Literal['hf-inference']="hf-inference"
@@ -40,11 +40,9 @@ async def get_agent():
                     }
                 }]
             )
-            print("🚀 Agent instance created successfully")
-            print("🔁 loading tools ...")
             await agent.load_tools()
             agent_instance = agent
-            print("✅ Tools loaded successfully")
+            print("✅ Agent is ready")
         except Exception as e:
             print(f"❌ Error creating/loading agent: {str(e)}")
     return agent_instance
@@ -77,15 +75,14 @@ def chat_function(user_message, history, model_id):
 
 
 def create_gradio_app():
-    with gr.Blocks(title="Model Card Chatbot",css=".gr-box{max-width: 800px; margin: auto;}") as demo:
+    with gr.Blocks(title="Model Card Chatbot") as demo:
         gr.Markdown("## 🤖 Model Card Chatbot\nAsk questions about Hugging Face model card")
-        with gr.Row():
-            model_id=gr.Textbox(label="MODEL ID", value="google/gemma-2-2b",scale=2)
-            user_input=gr.Textbox(label="Your Question",placeholder="Ask something about the model card .....",lines=1, scale=3)
-            send=gr.Button("Ask")
-            chatbot=gr.Chatbot(label="chat",height=50)
-            send.click(fn=chat_function, inputs=[user_input,chatbot,model_id], outputs=[chatbot,user_input])
-            return demo
+        model_id=gr.Textbox(label="MODEL ID", value="google/gemma-2-2b")
+        user_input=gr.Textbox(label="Your Question",placeholder="Ask something about the model card")
+        send=gr.Button("Ask")
+        chatbot=gr.Chatbot(label="chat")
+        send.click(fn=chat_function, inputs=[user_input,chatbot,model_id], outputs=[chatbot,user_input])
+    return demo
 gradio_app=create_gradio_app()
 app=gr.mount_gradio_app(app,gradio_app,path="/")