Spaces:
Sleeping
Sleeping
mohammadreza pakzadian commited on
Commit ·
cb21c73
1
Parent(s): 298ce6b
update model
Browse files
app.py
CHANGED
|
@@ -2,7 +2,7 @@ import os
|
|
| 2 |
import gradio as gr
|
| 3 |
import requests
|
| 4 |
import pandas as pd
|
| 5 |
-
from smolagents import CodeAgent, OpenAIServerModel
|
| 6 |
from tools.final_answer import FinalAnswerTool
|
| 7 |
from tools.visit_webpage import VisitWebpageTool
|
| 8 |
from tools.web_search import DuckDuckGoSearchTool
|
|
@@ -19,7 +19,12 @@ class BasicAgent:
|
|
| 19 |
final_answer = FinalAnswerTool()
|
| 20 |
visit_webpage = VisitWebpageTool()
|
| 21 |
web_search = DuckDuckGoSearchTool()
|
| 22 |
-
model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 23 |
self.agent = CodeAgent(
|
| 24 |
model=model,
|
| 25 |
tools=[visit_webpage, web_search, final_answer],
|
|
|
|
| 2 |
import gradio as gr
|
| 3 |
import requests
|
| 4 |
import pandas as pd
|
| 5 |
+
from smolagents import CodeAgent, OpenAIServerModel, HfApiModel
|
| 6 |
from tools.final_answer import FinalAnswerTool
|
| 7 |
from tools.visit_webpage import VisitWebpageTool
|
| 8 |
from tools.web_search import DuckDuckGoSearchTool
|
|
|
|
| 19 |
final_answer = FinalAnswerTool()
|
| 20 |
visit_webpage = VisitWebpageTool()
|
| 21 |
web_search = DuckDuckGoSearchTool()
|
| 22 |
+
# model = OpenAIServerModel(model_id="gpt-4.1", api_key=os.getenv("OPENAI_API_KEY"))
|
| 23 |
+
model = HfApiModel(
|
| 24 |
+
max_tokens=2096,
|
| 25 |
+
temperature=0.5,
|
| 26 |
+
model_id='Qwen/Qwen2.5-Coder-7B-Instruct', custom_role_conversions=None,
|
| 27 |
+
)
|
| 28 |
self.agent = CodeAgent(
|
| 29 |
model=model,
|
| 30 |
tools=[visit_webpage, web_search, final_answer],
|