Update app.py

app.py

@@ -13,55 +13,55 @@ model = ORTModelForSeq2SeqLM.from_pretrained(
 tokenizer = AutoTokenizer.from_pretrained("Genn9508/Sss")
 
 # Пайплайн
-pipe = pipeline("text2text-generation", model=model, tokenizer=tokenizer)
+pipe = pipeline("summarization", model=model, tokenizer=tokenizer)
 
-# Функция генерации
-def generate(text, task):
-    if task == "Сделать кратко":
-        prompt = f"сделай кратко: {text}"
-    elif task == "Перевести на английский":
-        prompt = f"переведи на английский: {text}"
-    elif task == "Перефразировать":
-        prompt = f"перефразируй: {text}"
-    else:
-        prompt = text
-
-    result = pipe(prompt, max_length=100, min_length=10, do_sample=False, no_repeat_ngram_size=2)
-    return result[0]['generated_text']
+# Функция суммаризации
+def summarize(text):
+    if not text.strip():
+        return "Введите текст для краткого изложения."
+    
+    # Генерация краткого варианта
+    result = pipe(
+        text,
+        max_length=100,      # максимум слов в выводе
+        min_length=20,       # минимум — чтобы не было слишком коротко
+        do_sample=False,     # детерминированная генерация
+        no_repeat_ngram_size=2,
+        truncation=True      # если текст длиннее 512 токенов
+    )
+    return result[0]['summary_text']
 
 # Интерфейс
 with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 🚀 T5 ONNX INT8 — Sss by Genn9508")
-    gr.Markdown("Модель оптимизирована для быстрой генерации: суммаризация, перевод, перефразирование.")
-
-    with gr.Row():
-        task = gr.Radio(
-            ["Сделать кратко", "Перевести на английский", "Перефразировать"],
-            label="Выберите задачу",
-            value="Сделать кратко"
-        )
+    gr.Markdown("# 📝 ONNX T5 — Суммаризатор текста")
+    gr.Markdown("Модель оптимизирована для **краткого изложения текста**. Просто введите любой текст — и получите краткую версию.")
 
     with gr.Row():
         with gr.Column(scale=2):
             input_text = gr.Textbox(
-                label="Введите текст",
-                placeholder="Например: Искусственный интеллект меняет все сферы жизни...",
-                lines=5
+                label="Текст для краткого изложения",
+                placeholder="Введите текст (статью, отзыв, письмо и т.д.) — модель сделает кратко.",
+                lines=6
             )
-            btn = gr.Button("🚀 Обработать", variant="primary")
+            btn = gr.Button("✨ Сделать кратко", variant="primary")
 
         with gr.Column(scale=2):
-            output_text = gr.Textbox(label="Результат", lines=5, interactive=False)
+            output_text = gr.Textbox(
+                label="Краткое изложение",
+                lines=6,
+                interactive=False
+            )
 
-    btn.click(fn=generate, inputs=[input_text, task], outputs=output_text)
+    btn.click(fn=summarize, inputs=input_text, outputs=output_text)
 
     gr.Examples(
         [
-            ["Сделай кратко", "Искусственный интеллект развивается быстрыми темпами и влияет на образование, медицину и транспорт."],
-            ["Перевести на английский", "Привет Как дела? Сегодня отличная погода."],
-            ["Перефразировать", "Этот фильм очень интересный и захватывающий."]
+            ["Искусственный интеллект меняет мир. Он используется в медицине, образовании, транспорте и других сферах. Благодаря ИИ люди могут решать сложные задачи быстрее и точнее."],
+            ["Компания запустила новый продукт. Он помогает автоматизировать рутинные задачи. Пользователи уже оставили положительные отзывы."],
+            ["На прошлой неделе прошла конференция по искусственному интеллекту. Участники обсудили последние достижения в области генеративных моделей и этики ИИ."]
         ],
-        inputs=[task, input_text]
+        inputs=input_text,
+        label="Примеры текстов"
     )
 
 demo.launch()