Update app.py

app.py

@@ -11,9 +11,11 @@ pipeline = Pipeline.from_pretrained(
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 pipeline.to(device)
 
+def process(audio):
+    if audio is None:
+        return {"error": "No audio"}
 
-def diarize(audio_path):
-    diarization = pipeline(audio_path)
+    diarization = pipeline(audio)
 
     segments = []
     speaker_map = {}
@@ -28,25 +30,7 @@ def diarize(audio_path):
             "end": round(turn.end, 2)
         })
 
-    # merge continuous segments
-    merged = []
-    if segments:
-        current = segments[0]
-        for seg in segments[1:]:
-            if seg["speaker"] == current["speaker"] and seg["start"] <= current["end"] + 0.3:
-                current["end"] = seg["end"]
-            else:
-                merged.append(current)
-                current = seg
-        merged.append(current)
-
-    return {"segments": merged}
-
-
-def process(audio):
-    if audio is None:
-        return {"error": "No audio"}
-    return diarize(audio)
+    return {"segments": segments}
 
 
 demo = gr.Interface(