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"]