Update app.py
Browse files
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 |
|