claudi47 commited on
Commit
593495d
·
1 Parent(s): 2226feb

specified the LLM model

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -17,7 +17,10 @@ class BasicAgent:
17
  def __init__(self):
18
  print("BasicAgent initialized.")
19
  search_tool = DuckDuckGoSearchTool()
20
- model = InferenceClientModel()
 
 
 
21
  self.agent = CodeAgent(
22
  model=model,
23
  tools=[search_tool],
 
17
  def __init__(self):
18
  print("BasicAgent initialized.")
19
  search_tool = DuckDuckGoSearchTool()
20
+ model = InferenceClientModel(
21
+ model_id="Qwen/Qwen2.5-Coder-32B-Instruct",
22
+ token=os.getenv("HF_TOKEN"),
23
+ )
24
  self.agent = CodeAgent(
25
  model=model,
26
  tools=[search_tool],