Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -30,12 +30,12 @@ class BasicAgent:
30
  # custom_role_conversions=None,
31
  #)
32
 
33
- #model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-0.5B-Instruct",
34
- # max_tokens=2096,
35
- # token=token,
36
- # custom_role_conversions=None)
37
  self.agent = CodeAgent(
38
- model = HfApiModel(),
39
  tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
40
  #additional_authorized_imports = ["pandas", "numpy"]
41
  )
 
30
  # custom_role_conversions=None,
31
  #)
32
 
33
+ model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-0.5B-Instruct",
34
+ max_tokens=2096,
35
+ token=token,
36
+ custom_role_conversions=None)
37
  self.agent = CodeAgent(
38
+ model = model,
39
  tools = [DuckDuckGoSearchTool(), WikipediaSearchTool(), VisitWebpageTool()],
40
  #additional_authorized_imports = ["pandas", "numpy"]
41
  )