update

app.py

@@ -1,34 +1,26 @@
 import gradio as gr
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
+from transformers import pipeline
 
-# Load model
-model_name = "Manish014/review-summariser-gpt-config1"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+pipe = pipeline("summarization", model="Manish014/review-summariser-gpt-config1")
 
-summarizer = pipeline("summarization", model=model, tokenizer=tokenizer)
-
-# Define Gradio UI
-def summarize_review(review):
-    if not review.strip():
+def summarize(text):
+    if not text.strip():
         return "Please enter a review."
-    output = summarizer(review, max_length=60, min_length=10, do_sample=False)
-    return output[0]["summary_text"]
+    return pipe(text)[0]["summary_text"]
 
-examples = [
-    ["This is the worst coffee maker I’ve ever used. It leaks water everywhere, the buttons barely work, and the brew tastes like plastic."],
-    ["Great product! Easy to use, super fast delivery, and the quality is outstanding. Highly recommend!"],
-    ["Battery life is average but display quality and sound are top-notch."],
-    ["Not worth the money. Poor customer service and the build feels cheap."]
+example_reviews = [
+    ["This product leaks water and smells like burnt plastic."],
+    ["Absolutely loved the screen resolution and battery life."],
+    ["Worst purchase I've made. Do not recommend at all."]
 ]
 
 demo = gr.Interface(
-    fn=summarize_review,
-    inputs=gr.Textbox(lines=6, placeholder="Paste a product review here..."),
-    outputs="text",
-    examples=examples,
-    title="Review Summariser GPT",
-    description="Enter a product review to get a helpful summary using Config1 fine-tuned T5-small model."
+    fn=summarize,
+    inputs=gr.Textbox(lines=5, placeholder="Enter product review..."),
+    outputs=gr.Textbox(label="Summary"),
+    examples=example_reviews,
+    title="Review Summariser GPT - Config 1",
+    description="Enter a long product review and get a helpful short summary."
 )
 
 demo.launch()