Project Alpha 3

2024-10-10 11:03:15 -04:00
parent 1650f3fccb
commit d3e07ede69
2 changed files with 8 additions and 1 deletions
--- a/summarize.py
+++ b/summarize.py
@@ -1,6 +1,12 @@
 from transformers import pipeline

 def process(input_text):
-    summarizer = pipeline("summarization", model="marianna13/flan-t5-base-summarization", max_length=200, min_length=10, device="mps")
+    summarizer = pipeline(
+        "summarization",
+        model="marianna13/flan-t5-base-summarization",
+        max_length=200,
+        min_length=10,
+        device="mps"
+    )
    output = summarizer(input_text)[0]['summary_text']
    return output