Update app.py

app.py

@@ -1,49 +1,45 @@
 import gradio as gr
 from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
+import torch
 
-# 1. Configuration
+# 1. Setup
 MODEL_ID = "VoltIC/Automated-Text-Summarizer"
 SUBFOLDER = "summarizer_model"
 
-print("🚀 Loading model and tokenizer...")
+# 2. Initialization (No 'try' block here so we can see the real error in logs)
+print("🚀 Phase 1: Loading Tokenizer...")
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, subfolder=SUBFOLDER)
 
-try:
-    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID, subfolder=SUBFOLDER)
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_ID, subfolder=SUBFOLDER)
+print("🚀 Phase 2: Loading Model Weights (1.6GB)...")
+model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_ID, subfolder=SUBFOLDER)
 
-    summarizer = pipeline(
-        "summarization", 
-        model=model, 
-        tokenizer=tokenizer,
-        framework="pt"
-    )
-    print("✅ System Ready")
-except Exception as e:
-    print(f"❌ Initialization Error: {e}")
+print("🚀 Phase 3: Building Pipeline...")
+# This defines the 'summarizer' variable globally
+summarizer = pipeline(
+    "summarization", 
+    model=model, 
+    tokenizer=tokenizer,
+    framework="pt",
+    device=-1 # Forces CPU
+)
+print("✅ SUCCESS: Summarizer is now defined!")
 
+# 3. Logic
 def summarize_text(text):
-    if not text or len(text.strip()) < 20:
-        return "Error: Text is too short."
+    if not text:
+        return "Please enter text."
     
-    try:
-        result = summarizer(
-            text, 
-            max_length=100, 
-            min_length=30, 
-            do_sample=False,
-            truncation=True
-        )
-        return result[0]['summary_text']
-    except Exception as e:
-        return f"Runtime Error: {str(e)}"
+    # We use the global 'summarizer' defined above
+    results = summarizer(text, max_length=100, min_length=30, truncation=True)
+    return results[0]['summary_text']
 
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
-    gr.Markdown("# 📝 AI Text Summarizer")
-    with gr.Column():
-        input_box = gr.Textbox(label="Input Text", lines=8)
-        output_box = gr.Textbox(label="Summary Result", lines=4)
-        btn = gr.Button("Summarize", variant="primary")
-    
-    btn.click(fn=summarize_text, inputs=input_box, outputs=output_box)
+# 4. Interface
+demo = gr.Interface(
+    fn=summarize_text,
+    inputs=gr.Textbox(lines=10, label="Input Text"),
+    outputs=gr.Textbox(label="Summary"),
+    title="AI Text Summarizer"
+)
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()