Update app.py

app.py

@@ -16,17 +16,20 @@ pipe = pipeline(
 )
 
 def transcribe(file, task, return_timestamps):
-    outputs = pipe(file, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=return_timestamps)
+    outputs = pipe(file, batch_size=BATCH_SIZE, generate_kwargs={"task": task}, return_timestamps=True)
     text = outputs["text"]
-    if return_timestamps:
-        timestamps = outputs["chunks"]
-        timestamps = [
-            f"[{format_timestamp(chunk['timestamp'][0])} -> {format_timestamp(chunk['timestamp'][1])}] {chunk['text']}"
-            for chunk in timestamps
-        ]
-        text = "\n".join(str(feature) for feature in timestamps)
+    timestamps = outputs["chunks"]
+    
+    if return_timestamps==True:
+      timestamps = [f"[{format_timestamp(chunk['timestamp'][0])} -> {format_timestamp(chunk['timestamp'][1])}] {chunk['text']}" for chunk in timestamps]
+      
+    else:
+      timestamps = [f"{chunk['text']}" for chunk in timestamps]
+      
+    text = "\n".join(str(feature) for feature in timestamps)    
     return text
 
+
 file_transcribe = gr.Interface(
     fn=transcribe,
     inputs=[
@@ -48,4 +51,4 @@ file_transcribe = gr.Interface(
     allow_flagging="never",
 )
 
-file_transcribe.launch(enable_queue=True)
+file_transcribe.launch(enable_queue=True, debug = True)