artyomboyko commited on
Commit
96bd470
·
verified ·
1 Parent(s): b1066a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
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, DuckDuckGoSearchTool, TransformersModel
 
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="HuggingFaceTB/SmolLM2-1.7B-Instruct", device_map="auto", max_new_tokens=1000)
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