mmichiels13 commited on
Commit
4c13ada
·
verified ·
1 Parent(s): 9af71f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -42,8 +42,8 @@ class BasicAgent:
42
  # self.search_item_ctrl_f = search_item_ctrl_f
43
 
44
  # Create the model
45
- self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
46
- # self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct") #smaller/faster
47
  # self.model = OpenAIServerModel(model_id="gpt-4o")
48
 
49
  # Define your system prompt
@@ -54,7 +54,7 @@ class BasicAgent:
54
  tools=[self.web_search, self.visit_webpage, self.final_answer],
55
  # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
56
  model=self.model,
57
- additional_authorized_imports=["pandas", "openpyxl", "yt_dlp", "requests", "io", "json"]
58
  max_steps=10 # Limit reasoning steps
59
  )
60
 
 
42
  # self.search_item_ctrl_f = search_item_ctrl_f
43
 
44
  # Create the model
45
+ # self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-72B-Instruct")
46
+ self.model = InferenceClientModel(model_id="Qwen/Qwen2.5-7B-Instruct") #smaller/faster
47
  # self.model = OpenAIServerModel(model_id="gpt-4o")
48
 
49
  # Define your system prompt
 
54
  tools=[self.web_search, self.visit_webpage, self.final_answer],
55
  # tools=[self.web_search, self.visit_webpage, self.final_answer, go_back, close_popups, search_item_ctrl_f],
56
  model=self.model,
57
+ additional_authorized_imports=["pandas", "openpyxl", "yt_dlp", "requests", "io", "json"],
58
  max_steps=10 # Limit reasoning steps
59
  )
60