funcky commited on
Commit
e520190
·
verified ·
1 Parent(s): b9c1b5f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -48,8 +48,8 @@ final_answer = FinalAnswerTool()
48
  model = HfApiModel(
49
  max_tokens=2096,
50
  temperature=0.5,
51
- # model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
52
- model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
53
  custom_role_conversions=None,
54
  )
55
 
@@ -62,7 +62,7 @@ with open("prompts.yaml", 'r') as stream:
62
 
63
  agent = CodeAgent(
64
  model=model,
65
- tools=[my_custom_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
66
  max_steps=6,
67
  verbosity_level=1,
68
  grammar=None,
 
48
  model = HfApiModel(
49
  max_tokens=2096,
50
  temperature=0.5,
51
+ model_id='https://wxknx1kg971u7k1n.us-east-1.aws.endpoints.huggingface.cloud',# it is possible that this model may be overloaded
52
+ # model_id="deepseek-ai/DeepSeek-R1-Distill-Qwen-32B",
53
  custom_role_conversions=None,
54
  )
55
 
 
62
 
63
  agent = CodeAgent(
64
  model=model,
65
+ tools=[random_tool, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
66
  max_steps=6,
67
  verbosity_level=1,
68
  grammar=None,