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 |
|