MacKov commited on
Commit
2d17a67
·
verified ·
1 Parent(s): 9b90756

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -7
app.py CHANGED
@@ -20,9 +20,7 @@ DEFAULT_API_URL = "https://agents-course-unit4-scoring.hf.space"
20
  MODEL_ID = "Qwen/Qwen2.5-32B-Instruct"
21
 
22
  class BasicAgent:
23
- def __init__(self):
24
- print("Инициализация агента...")
25
-
26
  self.model = InferenceClientModel(
27
  model_id=MODEL_ID,
28
  token=os.getenv("HF_TOKEN"),
@@ -111,11 +109,9 @@ class BasicAgent:
111
  def __call__(self, question: str) -> str:
112
  print(f"Вопрос: {question[:120]}...")
113
 
114
- # Обрезка
115
  if len(question) > 2500:
116
  question = question[:2500] + "\n[Обрезано из-за длины.]"
117
 
118
- # Хак для файлов/видео/аудио
119
  q = question.lower()
120
  if any(k in q for k in [".mp3", "audio", "recording", "voice", "youtube.com", "video", "attached", "file", "excel", "pdf", "image"]):
121
  question += "\nЕсли есть URL — скачай файл/видео и читай/ищи транскрипт/описание. Не придумывай."
@@ -124,7 +120,6 @@ class BasicAgent:
124
  result = self.agent.run(question)
125
  answer = str(result).strip()
126
 
127
- # Жёсткая очистка
128
  prefixes = [
129
  "Final Answer", "Final answer", "Answer:", "The answer is",
130
  "So the final answer is", "```", "boxed{", "}", "[/INST]", "</s>",
@@ -139,7 +134,7 @@ class BasicAgent:
139
  answer = answer[1:-1].strip()
140
 
141
  answer = answer.strip()
142
- # Если ответ слишком длинный или странный — обрезаем
143
  if len(answer) > 300 or "придум" in answer.lower():
144
  answer = answer[:150] + "..." if len(answer) > 150 else answer
145
 
 
20
  MODEL_ID = "Qwen/Qwen2.5-32B-Instruct"
21
 
22
  class BasicAgent:
23
+ def __init__(self):
 
 
24
  self.model = InferenceClientModel(
25
  model_id=MODEL_ID,
26
  token=os.getenv("HF_TOKEN"),
 
109
  def __call__(self, question: str) -> str:
110
  print(f"Вопрос: {question[:120]}...")
111
 
 
112
  if len(question) > 2500:
113
  question = question[:2500] + "\n[Обрезано из-за длины.]"
114
 
 
115
  q = question.lower()
116
  if any(k in q for k in [".mp3", "audio", "recording", "voice", "youtube.com", "video", "attached", "file", "excel", "pdf", "image"]):
117
  question += "\nЕсли есть URL — скачай файл/видео и читай/ищи транскрипт/описание. Не придумывай."
 
120
  result = self.agent.run(question)
121
  answer = str(result).strip()
122
 
 
123
  prefixes = [
124
  "Final Answer", "Final answer", "Answer:", "The answer is",
125
  "So the final answer is", "```", "boxed{", "}", "[/INST]", "</s>",
 
134
  answer = answer[1:-1].strip()
135
 
136
  answer = answer.strip()
137
+
138
  if len(answer) > 300 or "придум" in answer.lower():
139
  answer = answer[:150] + "..." if len(answer) > 150 else answer
140