Update document-based-assistant/app.py

document-based-assistant/app.py

@@ -47,16 +47,27 @@ agent = CodeAgent(
     stream_outputs=True
 )
 
-def agent_interface(query):
+def predict(query: str):
+    """GAIA entrypoint"""
     return agent.run(query)
 
 iface = gr.Interface(
-    fn=agent_interface,
+    fn=predict,
     inputs="text",
     outputs="text",
     title="Document QA Agent"
 )
 
-# if __name__ == "__main__":
-#     iface.launch(server_name="0.0.0.0", server_port=7860)
+# def agent_interface(query):
+#     return agent.run(query)
+
+# iface = gr.Interface(
+#     fn=agent_interface,
+#     inputs="text",
+#     outputs="text",
+#     title="Document QA Agent"
+# )
+
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", server_port=7860)