boompack commited on
Commit
1f8811b
1 Parent(s): 37f85d9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -7
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
- # Пример генерации текста с помощью GPT-Neo
22
- gpt_result = self.model_gpt(text, max_length=150) # Увеличили max_length
23
- return gpt_result
 
 
 
 
24
  elif step == 2:
25
- # Пример генерации текста с помощью T5 (например, для резюмирования)
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