From d3e07ede69ddb1125fa9fa620a63b7cd766ce09f Mon Sep 17 00:00:00 2001
From: Elizabeth Cray <elizabeth.cray@afs.com>
Date: Thu, 10 Oct 2024 11:03:15 -0400
Subject: [PATCH] Project Alpha 3

---
 summarize.py  | 8 +++++++-
 transcribe.py | 1 +
 2 files changed, 8 insertions(+), 1 deletion(-)

diff --git a/summarize.py b/summarize.py
index c806873..0d9901d 100644
--- 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
diff --git a/transcribe.py b/transcribe.py
index 0b2027e..c75b5ee 100644
--- a/transcribe.py
+++ b/transcribe.py
@@ -15,6 +15,7 @@ def process(audio_path):
         tokenizer=processor.tokenizer,
         feature_extractor=processor.feature_extractor,
         torch_dtype=torch.float32,
+        return_timestamps=True,
         device="mps"
     )
     output = pipe(audio_path)["text"]