Update app.py
Browse files
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
|