Project Alpha 1

2024-10-10 10:51:48 -04:00
parent c03af29cc6
commit 120c1f7abc
7 changed files with 157 additions and 0 deletions
--- a/summarize.py
+++ b/summarize.py
@@ -0,0 +1,6 @@
+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")
+    output = summarizer(input_text)[0]['summary_text']
+    return output