Update app.py
Browse files
app.py
CHANGED
@@ -4,10 +4,14 @@ import requests
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
|
7 |
-
from smolagents import load_tool, CodeAgent, InferenceClientModel,
|
|
|
8 |
from smolagents.agents import CodeAgent, ToolCallingAgent
|
9 |
from dotenv import load_dotenv
|
10 |
|
|
|
|
|
|
|
11 |
# (Сохраните константы как есть)
|
12 |
# --- Константы ---
|
13 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
@@ -39,8 +43,8 @@ class BasicAgent:
|
|
39 |
# planning_interval=3
|
40 |
# )
|
41 |
|
42 |
-
self.model = TransformersModel(model_id=
|
43 |
-
self.agent = ToolCallingAgent(tools=[search_tool], model=self.model, verbosity_level=2)
|
44 |
|
45 |
print("Agent initialized successfully.")
|
46 |
|
|
|
4 |
import inspect
|
5 |
import pandas as pd
|
6 |
|
7 |
+
from smolagents import load_tool, CodeAgent, InferenceClientModel, TransformersModel
|
8 |
+
from smolagents import DuckDuckGoSearchTool, VisitWebpageTool, SpeechToTextTool
|
9 |
from smolagents.agents import CodeAgent, ToolCallingAgent
|
10 |
from dotenv import load_dotenv
|
11 |
|
12 |
+
# --- Настраиваемые параметры
|
13 |
+
hf_model = "google/gemma-3-4b-it"
|
14 |
+
|
15 |
# (Сохраните константы как есть)
|
16 |
# --- Константы ---
|
17 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
|
|
43 |
# planning_interval=3
|
44 |
# )
|
45 |
|
46 |
+
self.model = TransformersModel(model_id=hf_model, device_map="auto", max_new_tokens=1000)
|
47 |
+
self.agent = ToolCallingAgent(tools=[search_tool, visit_page_tool, speech_to_text_tool], model=self.model, verbosity_level=2)
|
48 |
|
49 |
print("Agent initialized successfully.")
|
50 |
|