Update app.py

app.py

@@ -1,23 +1,36 @@
-import gradio as gr
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+import gradio as gr
 
-# Load the model and tokenizer
+# Load model and tokenizer
 tokenizer = AutoTokenizer.from_pretrained("mynkchaudhry/Summarization-Pro")
 model = AutoModelForSeq2SeqLM.from_pretrained("mynkchaudhry/Summarization-Pro")
 
 def summarize(text):
-    inputs = tokenizer.encode("summarize: " + text, return_tensors="pt", max_length=1024, truncation=True)
-    summary_ids = model.generate(inputs, max_length=800, min_length=300, length_penalty=2.0, num_beams=4, early_stopping=True)
+    inputs = tokenizer(text, return_tensors="pt", max_length=1024, truncation=True)
+    summary_ids = model.generate(
+        inputs.input_ids,
+        max_length=1500,  # Increased max_length for longer summaries
+        min_length=100,   # Ensures summaries are sufficiently long
+        length_penalty=0.9,  # Encourages longer summaries but balances precision
+        num_beams=6,      # More beams for better exploration of possible summaries
+        early_stopping=True
+    )
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary
 
-# Create a Gradio interface
-interface = gr.Interface(theme="NoCrypt/miku",
-                         fn=summarize, 
-                         inputs=gr.Textbox(lines=10, label="Input Text"), 
-                         outputs=gr.Textbox(label="Summary"),
-                         title="Text Summarization Tool",
-                         description="Enter a long text and get a summarized version of it.")
+# Gradio interface
+interface = gr.Interface(
+    fn=summarize,
+    inputs=gr.Textbox(lines=10, label="Input Text"),
+    outputs=gr.Textbox(label="Summary"),
+    examples=[
+        ["The quick brown fox jumps over the lazy dog. This sentence is an English-language pangram—a sentence that contains all the letters of the English alphabet. It is often used for testing typewriters, keyboards, and computer fonts."],
+        ["Artificial intelligence (AI) is intelligence demonstrated by machines, in contrast to the natural intelligence displayed by humans and animals. Leading AI textbooks define the field as the study of intelligent agents: any device that perceives its environment and takes actions that maximize its chance of successfully achieving its goals."],
+        ["In data science, an anomaly is an observation that deviates significantly from other observations. Anomalies are also known as outliers, novelties, noise, deviations, and exceptions. Anomaly detection finds application in many domains, including fraud detection, network security, and fault detection."]
+    ],
+    title="Text Summarization",
+    description="Generate a summary from the input text using a pre-trained summarization model.",
+    theme="freddyaboulton/dracula_revamped"  # Ensure this theme exists or use a default theme
+)
 
-# Launch the interface
 interface.launch()