Update app.py
Browse files
app.py
CHANGED
@@ -17,14 +17,7 @@ hf_model = "google/gemma-3-4b-it"
|
|
17 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
18 |
|
19 |
# --- Инициализация ---
|
20 |
-
|
21 |
-
if HF_TOKEN:
|
22 |
-
login(token=os.getenv("HF_ACCESS_TOKEN"))
|
23 |
-
print("Успешный вход в Hugging Face Hub.")
|
24 |
-
else:
|
25 |
-
print("HF_ACCESS_TOKEN не найден. Запуск в оффлайн-режиме или с публичными моделями.")
|
26 |
-
except Exception as e:
|
27 |
-
print(f"Не удалось войти в Hugging Face Hub: {e}")
|
28 |
|
29 |
search_tool = DuckDuckGoSearchTool()
|
30 |
# ----------------------
|
@@ -32,13 +25,19 @@ search_tool = DuckDuckGoSearchTool()
|
|
32 |
# --- Основное определение агента ---
|
33 |
# ----- ЗДЕСЬ ВЫ МОЖЕТЕ СОЗДАТЬ ТО, ЧТО ХОТИТЕ ------
|
34 |
class BasicAgent:
|
|
|
35 |
def __init__(self):
|
36 |
print("BasicAgent initialized.")
|
37 |
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
|
|
|
|
|
|
|
|
|
|
42 |
|
43 |
self.model = TransformersModel(model_id=hf_model, device_map="auto", max_new_tokens=1000)
|
44 |
self.agent = ToolCallingAgent(tools=[search_tool, visit_page_tool, speech_to_text_tool], model=self.model, verbosity_level=2)
|
|
|
17 |
DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
|
18 |
|
19 |
# --- Инициализация ---
|
20 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
21 |
|
22 |
search_tool = DuckDuckGoSearchTool()
|
23 |
# ----------------------
|
|
|
25 |
# --- Основное определение агента ---
|
26 |
# ----- ЗДЕСЬ ВЫ МОЖЕТЕ СОЗДАТЬ ТО, ЧТО ХОТИТЕ ------
|
27 |
class BasicAgent:
|
28 |
+
|
29 |
def __init__(self):
|
30 |
print("BasicAgent initialized.")
|
31 |
|
32 |
+
try:
|
33 |
+
if HF_TOKEN:
|
34 |
+
login(token=os.getenv("HF_ACCESS_TOKEN"))
|
35 |
+
print("Успешный вход в Hugging Face Hub.")
|
36 |
+
else:
|
37 |
+
print("HF_ACCESS_TOKEN не найден. Запуск в оффлайн-режиме или с публичными моделями.")
|
38 |
+
except Exception as e:
|
39 |
+
print(f"Не удалось войти в Hugging Face Hub: {e}")
|
40 |
+
|
41 |
|
42 |
self.model = TransformersModel(model_id=hf_model, device_map="auto", max_new_tokens=1000)
|
43 |
self.agent = ToolCallingAgent(tools=[search_tool, visit_page_tool, speech_to_text_tool], model=self.model, verbosity_level=2)
|