Ali2206 commited on
Commit
34915cc
·
verified ·
1 Parent(s): 6dea94e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -23,8 +23,7 @@ report_dir = os.path.join(persistent_dir, "reports")
23
  for directory in [model_cache_dir, tool_cache_dir, file_cache_dir, report_dir]:
24
  os.makedirs(directory, exist_ok=True)
25
 
26
- os.environ["HF_HOME"] = model_cache_dir
27
- os.environ["TRANSFORMERS_CACHE"] = model_cache_dir
28
 
29
  current_dir = os.path.dirname(os.path.abspath(__file__))
30
  src_path = os.path.abspath(os.path.join(current_dir, "src"))
@@ -134,8 +133,7 @@ def init_agent():
134
  enable_checker=True,
135
  step_rag_num=4,
136
  seed=100,
137
- additional_default_tools=[],
138
- max_model_tokens=MAX_MODEL_TOKENS # Pass the updated token limit
139
  )
140
  agent.init_model()
141
  return agent
 
23
  for directory in [model_cache_dir, tool_cache_dir, file_cache_dir, report_dir]:
24
  os.makedirs(directory, exist_ok=True)
25
 
26
+ os.environ["HF_HOME"] = model_cache_dir # Using HF_HOME instead of TRANSFORMERS_CACHE
 
27
 
28
  current_dir = os.path.dirname(os.path.abspath(__file__))
29
  src_path = os.path.abspath(os.path.join(current_dir, "src"))
 
133
  enable_checker=True,
134
  step_rag_num=4,
135
  seed=100,
136
+ additional_default_tools=[]
 
137
  )
138
  agent.init_model()
139
  return agent