Update app.py
Browse files
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 |
-
|
| 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 |
|