updating app.py

app.py

@@ -1,26 +1,47 @@
 import gradio as gr
 from transformers import pipeline
 
-pipe = pipeline("summarization", model="Manish014/review-summariser-gpt-config1")
+# Load model with custom generation settings
+pipe = pipeline(
+    "summarization",
+    model="Manish014/review-summariser-gpt-config1",
+    tokenizer="Manish014/review-summariser-gpt-config1",
+    device=0  # Use GPU if available, else set to -1
+)
 
+# Summarization function with settings
 def summarize(text):
     if not text.strip():
-        return "Please enter a review."
-    return pipe(text)[0]["summary_text"]
+        return "Please enter a product review."
+    
+    result = pipe(
+        text,
+        max_length=80,
+        min_length=10,
+        num_beams=4,
+        early_stopping=True,
+        length_penalty=1.2
+    )
+    return result[0]["summary_text"]
 
+# Example reviews
 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."]
 ]
 
+# Gradio Interface
 demo = gr.Interface(
     fn=summarize,
-    inputs=gr.Textbox(lines=5, placeholder="Enter product review..."),
+    inputs=gr.Textbox(label="Product Review", lines=5, placeholder="Enter product review..."),
     outputs=gr.Textbox(label="Summary"),
+    title="📝 Review Summariser GPT - Config 1",
+    description="Enter a long product review and get a helpful short summary.",
     examples=example_reviews,
-    title="Review Summariser GPT - Config 1",
-    description="Enter a long product review and get a helpful short summary."
+    theme="default"
 )
 
-demo.launch()
+# Launch app
+if __name__ == "__main__":
+    demo.launch()