Burnside commited on
Commit
afc5fdc
·
verified ·
1 Parent(s): d67068f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -41,7 +41,8 @@ final_answer = FinalAnswerTool()
41
  model = HfApiModel(
42
  max_tokens=2096,
43
  temperature=0.5,
44
- model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded, fall back on 14B or even 7B or 3B locally
 
45
  custom_role_conversions=None,
46
  )
47
 
@@ -54,7 +55,7 @@ with open("prompts.yaml", 'r') as stream:
54
 
55
  agent = CodeAgent(
56
  model=model,
57
- tools=[image_generation_tool,get_current_time_in_timezone,final_answer], ## add your tools here (don't remove final answer)
58
  max_steps=6,
59
  verbosity_level=1,
60
  grammar=None,
 
41
  model = HfApiModel(
42
  max_tokens=2096,
43
  temperature=0.5,
44
+ #model_id='Qwen/Qwen2.5-Coder-32B-Instruct',# it is possible that this model may be overloaded, fall back on 14B or even 7B or 3B locally
45
+ model_id='Qwen/Qwen2.5-Coder-14B-Instruct',# it is possible that this model may be overloaded, fall back on 14B or even 7B or 3B locally
46
  custom_role_conversions=None,
47
  )
48
 
 
55
 
56
  agent = CodeAgent(
57
  model=model,
58
+ tools=[image_generation_tool, display_time, get_current_time_in_timezone, final_answer], ## add your tools here (don't remove final answer)
59
  max_steps=6,
60
  verbosity_level=1,
61
  grammar=None,