Update agent.py

agent.py

@@ -1,11 +1,13 @@
 from transformers import pipeline
+import time
+
 
 # Load once at startup
 classifier = pipeline(
     "zero-shot-classification",
     model="valhalla/distilbart-mnli-12-3"
 )
-import time
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
 def analyze_article(text, title, link):
     t0 = time.time()
@@ -14,8 +16,10 @@ def analyze_article(text, title, link):
         candidate_labels=["analytics", "data science", "business insight"],
         multi_label=True
     )
-    print(f"[DEBUG] Model inference time: {time.time() - t0:.2f}s")
-    summary = result['labels'][0]
+    print(f"[DEBUG] Classifier Model inference time: {time.time() - t0:.2f}s")
+     # Summarize article text
+    summary = summarizer(text[:1024], max_length=150, min_length=40, do_sample=False)[0]['summary_text']
+    print(f"[DEBUG] Summarizer Model inference time: {time.time() - t0:.2f}s")
     return {
         "title": title,
         "link": link,