ATK20 commited on
Commit
608fb36
·
verified ·
1 Parent(s): ce25683

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ class BasicAgent:
17
  print("BasicAgent initialized.")
18
  token_key = os.getenv("HFAPI_KEY")
19
  # Initialize the search tool
20
- model = HfApiModel(model_id="distilbert/distilgpt2", api_key=token_key)
21
  search_tool = DuckDuckGoSearchTool()
22
 
23
  self.agent = CodeAgent(model=model, tools=[search_tool])
 
17
  print("BasicAgent initialized.")
18
  token_key = os.getenv("HFAPI_KEY")
19
  # Initialize the search tool
20
+ model = HfApiModel(model_id="Qwen/Qwen2.5-VL-7B-Instruct", api_key=token_key)
21
  search_tool = DuckDuckGoSearchTool()
22
 
23
  self.agent = CodeAgent(model=model, tools=[search_tool])