Update app.py

app.py

@@ -30,7 +30,6 @@ current_dir = os.path.dirname(os.path.abspath(__file__))
 CONFIG = {
     "model_name": "mims-harvard/TxAgent-T1-Llama-3.1-8B",
     "rag_model_name": "mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
-    "embedding_filename": "ToolRAG-T1-GTE-Qwen2-1.5Btool_embedding.pt",
     "tool_files": {
         "opentarget": str(files('tooluniverse.data').joinpath('opentarget_tools.json')),
         "fda_drug_label": str(files('tooluniverse.data').joinpath('fda_drug_labeling_tools.json')),
@@ -53,22 +52,18 @@ def prepare_tool_files():
 
 def create_agent():
     prepare_tool_files()
-    try:
-        agent = TxAgent(
-            CONFIG["model_name"],
-            CONFIG["rag_model_name"],
-            tool_files_dict=CONFIG["tool_files"],
-            force_finish=True,
-            enable_checker=True,
-            step_rag_num=10,
-            seed=42,
-            additional_default_tools=["DirectResponse", "RequireClarification"]
-        )
-        agent.init_model()
-        return agent
-    except Exception as e:
-        logger.error(f"Agent initialization failed: {e}")
-        raise
+    agent = TxAgent(
+        CONFIG["model_name"],
+        CONFIG["rag_model_name"],
+        tool_files_dict=CONFIG["tool_files"],
+        force_finish=True,
+        enable_checker=True,
+        step_rag_num=10,
+        seed=42,
+        additional_default_tools=["DirectResponse", "RequireClarification"]
+    )
+    agent.init_model()
+    return agent
 
 def respond(msg, chat_history, temperature, max_new_tokens, max_tokens, multi_agent, conversation, max_round):
     if not isinstance(msg, str) or len(msg.strip()) <= 10:
@@ -131,7 +126,7 @@ def main():
     global agent
     agent = create_agent()
     demo = create_demo(agent)
-    demo.queue(concurrency_count=1, max_size=20).launch(share=True)
+    demo.queue().launch(share=True)
 
 if __name__ == "__main__":
-    main()
+    main()