Update app.py

app.py

@@ -1,4 +1,3 @@
-# Adapted from source [cite: 214-232]
 from fastapi import FastAPI, HTTPException
 from pydantic import BaseModel
 from transformers import pipeline
@@ -9,7 +8,7 @@ app = FastAPI()
 class TranscriptionPayload(BaseModel):
     text: str
 
-# Load the summarization pipeline on startup
+# Load summarizer on startup
 try:
     summarizer = pipeline(
         "summarization",
@@ -17,18 +16,15 @@ try:
         torch_dtype=torch.float32,
         device="cpu",
     )
+    print("✅ Summarizer model loaded successfully")
 except Exception as e:
     summarizer = None
-    print(f"Error loading summarization model: {e}")
+    print(f"❌ Error loading summarization model: {e}")
 
 @app.post("/summarize")
 async def summarize_text(payload: TranscriptionPayload):
     if not summarizer:
         raise HTTPException(status_code=503, detail="Summarizer model is not available.")
 
-    full_text = payload.text
-    
-    # Generate the summary
-    summary = summarizer(full_text, min_length=30, max_length=250, do_sample=False)
-    
-    return {"summary": summary[0]['summary_text']}
+    summary = summarizer(payload.text, min_length=30, max_length=250, do_sample=False)
+    return {"summary": summary[0]['summary_text']}