Update app/core/summarizer.py

app/core/summarizer.py

@@ -1,7 +1,10 @@
-from transformers import pipeline
-
-summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-
-def get_summary(text: str) -> str:
-    result = summarizer(text, max_length=150, min_length=30, do_sample=False)
-    return result[0]["summary_text"]
+import os
+from transformers import pipeline
+
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+
+summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
+def get_summary(text: str) -> str:
+    result = summarizer(text, max_length=150, min_length=30, do_sample=False)
+    return result[0]["summary_text"]