artyomboyko commited on
Commit
b370778
·
verified ·
1 Parent(s): feba2c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -8
app.py CHANGED
@@ -18,10 +18,23 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
18
 
19
  # --- Инициализация ---
20
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  search_tool = DuckDuckGoSearchTool()
 
 
23
  # ----------------------
24
 
 
25
  # --- Основное определение агента ---
26
  # ----- ЗДЕСЬ ВЫ МОЖЕТЕ СОЗДАТЬ ТО, ЧТО ХОТИТЕ ------
27
  class BasicAgent:
@@ -29,14 +42,7 @@ class BasicAgent:
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)
 
18
 
19
  # --- Инициализация ---
20
 
21
+ ACCESS_TOKEN=os.getenv("HF_ACCESS_TOKEN")
22
+
23
+ try:
24
+ if ACCESS_TOKEN:
25
+ login(token=ACCESS_TOKEN)
26
+ print("Успешный вход в Hugging Face Hub.")
27
+ else:
28
+ print("ACCESS_TOKEN не найден. Запуск в оффлайн-режиме или с публичными моделями.")
29
+ except Exception as e:
30
+ print(f"Не удалось войти в Hugging Face Hub: {e}")
31
 
32
  search_tool = DuckDuckGoSearchTool()
33
+ visit_page_tool = VisitWebpageTool()
34
+ speech_to_text_tool = SpeechToTextTool()
35
  # ----------------------
36
 
37
+
38
  # --- Основное определение агента ---
39
  # ----- ЗДЕСЬ ВЫ МОЖЕТЕ СОЗДАТЬ ТО, ЧТО ХОТИТЕ ------
40
  class BasicAgent:
 
42
  def __init__(self):
43
  print("BasicAgent initialized.")
44
 
45
+
 
 
 
 
 
 
 
46
 
47
 
48
  self.model = TransformersModel(model_id=hf_model, device_map="auto", max_new_tokens=1000)