Update app.py

app.py

@@ -5,6 +5,7 @@ import torch
 
 model_name = "Timur1984/t5-base-title-generation"   # тут конечно всем надо поставить свой акк и имя модели
 
+
 print("Loading model...")
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
@@ -27,20 +28,61 @@ 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(titles)
+    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):
 
+            text = gr.Textbox(
+                label="Article text",
+                lines=15,
+                placeholder="Paste article text here..."
+            )
 
-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"
-)
+            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.launch()
+demo.launch()