Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -30,13 +30,13 @@ class BasicAgent:
30
  # custom_role_conversions=None,
31
  #)
32
 
33
- model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-1.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
  )
42
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.
 
30
  # custom_role_conversions=None,
31
  #)
32
 
33
+ #model = InferenceClientModel(model_id="Qwen/Qwen2.5-Coder-1.5B-Instruct",
34
+ # max_tokens=2096,
35
+ # token=token,
36
+ # custom_role_conversions=None)
37
  self.agent = CodeAgent(
38
+ model = HfApiModel(),
39
+ tools = [DuckDuckGoSearchTool()],
40
  #additional_authorized_imports = ["pandas", "numpy"]
41
  )
42
  SYSTEM_PROMPT = """You are a general AI assistant. I will ask you a question.