Update app.py

app.py

@@ -5,7 +5,6 @@ import torch
 
 model_name = "Timur1984/t5-base-title-generation"   # тут конечно всем надо поставить свой акк и имя модели
 
-
 print("Loading model...")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
@@ -28,61 +27,20 @@ def generate_titles(text, num_titles, temperature):
     )
 
     decoded = tokenizer.batch_decode(outputs, skip_special_tokens=True)
-
     titles = [nltk.sent_tokenize(t.strip())[0] for t in decoded]
 
-    return "\n".join([f"• {t}" for t in titles])
-
-
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-
-    gr.Markdown(
-        """
-        # 📰 AI Article Title Generator
-        
-        Вставьте текст статьи — модель сгенерирует несколько вариантов заголовков.
-        """
-    )
-
-    with gr.Row():
-
-        with gr.Column(scale=3):
+    return "\n".join(titles)
 
-            text = gr.Textbox(
-                label="Article text",
-                lines=15,
-                placeholder="Paste article text here..."
-            )
 
-            generate_btn = gr.Button("Generate titles 🚀")
-
-        with gr.Column(scale=1):
-
-            num_titles = gr.Slider(
-                1,
-                10,
-                value=5,
-                step=1,
-                label="Number of titles"
-            )
-
-            temperature = gr.Slider(
-                0.1,
-                1.5,
-                value=0.7,
-                step=0.05,
-                label="Temperature"
-            )
-
-    output = gr.Textbox(
-        label="Generated titles",
-        lines=10
-    )
-
-    generate_btn.click(
-        fn=generate_titles,
-        inputs=[text, num_titles, temperature],
-        outputs=output
-    )
+interface = gr.Interface(
+    fn=generate_titles,
+    inputs=[
+        gr.Textbox(lines=15, label="Article text"),
+        gr.Slider(1,10,value=5,step=1,label="Number of titles"),
+        gr.Slider(0.1,1.5,value=0.7,step=0.05,label="Temperature")
+    ],
+    outputs="text",
+    title="Article Title Generator"
+)
 
-demo.launch()
+interface.launch()