Ali2206 commited on
Commit
354137b
·
verified ·
1 Parent(s): f7f7fed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -64,7 +64,6 @@ 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
- 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,8 +71,7 @@ def init_agent():
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
 
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
  force_finish=True,
72
  enable_checker=True,
73
  step_rag_num=4,
74
+ seed=100
 
75
  )
76
  agent.init_model()
77
  return agent