Add correct return

app.py

@@ -18,16 +18,17 @@ def pipeline(video_url):
     classification, probs = predict_from_document(sentences)
     #    return punctuated_text
     times, timestamps = match_mask_and_transcript(sentences, transcript, classification)
-    return [
-        {
-            "start": "12:05",
-            "end": "12:52",
-            "classification": str(classification),
-            "probabilities": probs,
-            "times": times,
-            "timestamps": timestamps,
-        }
-    ]
+    return [{"begin": time[0], "end": time[1]} for time in times]
+    #    return [
+    #        {
+    #            "start": "12:05",
+    #            "end": "12:52",
+    #            "classification": str(classification),
+    #            "probabilities": probs,
+    #            "times": times,
+    #            "timestamps": timestamps,
+    #        }
+    #    ]
 
 
 # print(pipeline("VL5M5ZihJK4"))