Update app.py

app.py

@@ -2,27 +2,23 @@ import gradio as gr
 import torch
 from parrot import Parrot
 
-# Check GPU/CPU
+# --- Model initialization ---
 device = "cuda" if torch.cuda.is_available() else "cpu"
 print(f"🧠 Using device: {device}")
 
-# Initialize model (ensure it's globally available)
 parrot = None
 try:
-    parrot = Parrot(
-        model_tag="prithivida/parrot_paraphraser_on_T5",
-        use_gpu=torch.cuda.is_available()
-    )
+    print("🔄 Loading Parrot model...")
+    parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=torch.cuda.is_available())
     print("✅ Parrot model loaded successfully.")
 except Exception as e:
-    print(f"❌ Error loading Parrot model: {e}")
+    print(f"❌ Model load failed: {str(e)}")
 
-# Define paraphrasing function
+# --- Paraphrasing function ---
 def paraphrase_text(input_text):
-    global parrot  # ensure it uses the global model
+    global parrot
     if not input_text.strip():
         return "⚠️ Please enter text to paraphrase."
-
     if parrot is None:
         return "❌ Model not loaded. Please restart the Space."
 
@@ -32,30 +28,21 @@ def paraphrase_text(input_text):
             diversity_ranker="levenshtein",
             do_diverse=True
         )
-
         if not paraphrases:
-            return "No paraphrases generated. Try another input."
+            return "No paraphrase generated. Try another input."
 
         result = ""
         for i, para_tuple in enumerate(paraphrases, start=1):
             result += f"{i}. {para_tuple[0]}\n\n"
         return result.strip()
-
     except Exception as e:
         return f"⚠️ Error during paraphrasing: {str(e)}"
 
-# Gradio interface
+# --- Interface ---
 interface = gr.Interface(
     fn=paraphrase_text,
-    inputs=gr.Textbox(
-        lines=3,
-        placeholder="Enter a short sentence (1–2 sentences max)",
-        label="Input Text"
-    ),
-    outputs=gr.Textbox(
-        lines=10,
-        label="Paraphrased Results"
-    ),
+    inputs=gr.Textbox(lines=3, placeholder="Enter 1–2 sentences", label="Input Text"),
+    outputs=gr.Textbox(lines=10, label="Paraphrased Results"),
     title="🦜 Parrot Paraphraser",
     description="Generate multiple diverse paraphrases using the Parrot T5 model. Ideal for rewording short sentences or phrases."
 )