Update app.py

app.py

@@ -96,8 +96,12 @@ def yt_transcribe(yt_url, task, max_filesize=75.0):
     inputs = {"array": inputs, "sampling_rate": pipe.feature_extractor.sampling_rate}
 
     text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)
-
-    return html_embed_str, text
+    text = pipe(inputs, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=return_timestamps)
+    
+    if return_timestamps:
+        return html_embed_str, chunks_to_srt(result['chunks'])
+    else:
+        return html_embed_str, result['text']
 
 
 demo = gr.Blocks()