Update app.py
Browse files
app.py
CHANGED
|
@@ -15,16 +15,18 @@ class TextProcessor:
|
|
| 15 |
|
| 16 |
def process_text(self, step, text):
|
| 17 |
"""
|
| 18 |
-
Обрабатывает текст с использованием выбранной модели.
|
| 19 |
"""
|
| 20 |
if step == 1:
|
| 21 |
-
#
|
| 22 |
-
|
| 23 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
elif step == 2:
|
| 25 |
-
#
|
| 26 |
-
t5_result = self.model_t5(text)
|
| 27 |
-
return t5_result
|
| 28 |
else:
|
| 29 |
return "Unknown step"
|
| 30 |
|
|
|
|
| 15 |
|
| 16 |
def process_text(self, step, text):
|
| 17 |
"""
|
| 18 |
+
Обрабатывает текст с использованием выбранной модели, разбивая текст на части.
|
| 19 |
"""
|
| 20 |
if step == 1:
|
| 21 |
+
# Разбиваем текст на части, например, по 500 символов
|
| 22 |
+
text_parts = [text[i:i + 500] for i in range(0, len(text), 500)]
|
| 23 |
+
gpt_results = []
|
| 24 |
+
for part in text_parts:
|
| 25 |
+
gpt_result = self.model_gpt(part, max_length=150)
|
| 26 |
+
gpt_results.append(gpt_result[0]['generated_text'])
|
| 27 |
+
return "\n".join(gpt_results) # Объединяем ответы для всех частей
|
| 28 |
elif step == 2:
|
| 29 |
+
# ... (код обработки с помощью T5)
|
|
|
|
|
|
|
| 30 |
else:
|
| 31 |
return "Unknown step"
|
| 32 |
|