Update app.py
Browse files
app.py
CHANGED
@@ -64,6 +64,7 @@ def init_agent():
|
|
64 |
if not os.path.exists(tool_path):
|
65 |
shutil.copy(os.path.abspath("data/new_tool.json"), tool_path)
|
66 |
|
|
|
67 |
agent = TxAgent(
|
68 |
model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
|
69 |
rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
|
@@ -71,7 +72,8 @@ def init_agent():
|
|
71 |
force_finish=True,
|
72 |
enable_checker=True,
|
73 |
step_rag_num=4,
|
74 |
-
seed=100
|
|
|
75 |
)
|
76 |
agent.init_model()
|
77 |
return agent
|
|
|
64 |
if not os.path.exists(tool_path):
|
65 |
shutil.copy(os.path.abspath("data/new_tool.json"), tool_path)
|
66 |
|
67 |
+
from txagent.txagent import TxAgent
|
68 |
agent = TxAgent(
|
69 |
model_name="mims-harvard/TxAgent-T1-Llama-3.1-8B",
|
70 |
rag_model_name="mims-harvard/ToolRAG-T1-GTE-Qwen2-1.5B",
|
|
|
72 |
force_finish=True,
|
73 |
enable_checker=True,
|
74 |
step_rag_num=4,
|
75 |
+
seed=100,
|
76 |
+
use_vllm=False # Disable vLLM for Hugging Face Spaces
|
77 |
)
|
78 |
agent.init_model()
|
79 |
return agent
|