vrishi10 commited on
Commit
e9d4354
·
verified ·
1 Parent(s): 29b4f57

update token and tools

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -6,6 +6,9 @@ import yaml
6
  from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
 
 
 
9
 
10
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
11
  @tool
@@ -67,6 +70,7 @@ model = HfApiModel(
67
  max_tokens=2096,
68
  temperature=0.5,
69
  model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
 
70
  custom_role_conversions=None,
71
  )
72
 
@@ -79,7 +83,7 @@ with open("prompts.yaml", 'r') as stream:
79
 
80
  agent = CodeAgent(
81
  model=model,
82
- tools=[final_answer], ## add your tools here (don't remove final answer)
83
  max_steps=6,
84
  verbosity_level=1,
85
  grammar=None,
 
6
  from tools.final_answer import FinalAnswerTool
7
 
8
  from Gradio_UI import GradioUI
9
+ import os
10
+
11
+
12
 
13
  # Below is an example of a tool that does nothing. Amaze us with your creativity !
14
  @tool
 
70
  max_tokens=2096,
71
  temperature=0.5,
72
  model_id='deepseek-ai/DeepSeek-R1-Distill-Qwen-32B',# it is possible that this model may be overloaded
73
+ token=os.getenv('hf_token'),
74
  custom_role_conversions=None,
75
  )
76
 
 
83
 
84
  agent = CodeAgent(
85
  model=model,
86
+ tools=[calculate,final_answer], ## add your tools here (don't remove final answer)
87
  max_steps=6,
88
  verbosity_level=1,
89
  grammar=None,