Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ import gradio as gr
|
|
| 3 |
import requests
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
| 6 |
-
from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, Tool, tool
|
| 7 |
|
| 8 |
# (Keep Constants as is)
|
| 9 |
# --- Constants ---
|
|
@@ -78,13 +78,14 @@ class BasicAgent:
|
|
| 78 |
self.agent = CodeAgent(
|
| 79 |
tools=[
|
| 80 |
DuckDuckGoSearchTool(),
|
|
|
|
| 81 |
fetch_task_file,
|
| 82 |
read_spreadsheet,
|
| 83 |
#os.environ["HF_TOKEN"] = "hf_token",
|
| 84 |
#image_analysis_tool,
|
| 85 |
],
|
| 86 |
model=InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"),
|
| 87 |
-
max_steps=
|
| 88 |
)
|
| 89 |
#answering questions
|
| 90 |
def __call__(self, question: str) -> str:
|
|
|
|
| 3 |
import requests
|
| 4 |
import inspect
|
| 5 |
import pandas as pd
|
| 6 |
+
from smolagents import CodeAgent, InferenceClientModel, DuckDuckGoSearchTool, Tool, tool, VisitWebpageTool
|
| 7 |
|
| 8 |
# (Keep Constants as is)
|
| 9 |
# --- Constants ---
|
|
|
|
| 78 |
self.agent = CodeAgent(
|
| 79 |
tools=[
|
| 80 |
DuckDuckGoSearchTool(),
|
| 81 |
+
VisitWebpageTool(),
|
| 82 |
fetch_task_file,
|
| 83 |
read_spreadsheet,
|
| 84 |
#os.environ["HF_TOKEN"] = "hf_token",
|
| 85 |
#image_analysis_tool,
|
| 86 |
],
|
| 87 |
model=InferenceClientModel("Qwen/Qwen2.5-Coder-32B-Instruct"),
|
| 88 |
+
max_steps=20,
|
| 89 |
)
|
| 90 |
#answering questions
|
| 91 |
def __call__(self, question: str) -> str:
|