Upload folder using huggingface_hub

asr_diarization/pipeline.py

@@ -66,16 +66,16 @@ class ASR_Diarization:
                 for word_info in result["chunks"]:
                     start_ts, end_ts = word_info.get("timestamp", (None, None)) or (None, None)
                     tokens.append({
+                        "tag": "w",
                         "start": start_ts,
                         "end": end_ts,
-                        "text": word_info["text"],
-                        "tag": "w"
+                        "text": word_info["text"]
                     })
 
             seg_dict = {
                 "speaker": spk,
-                "start": start,
-                "end": end,
+                "segment_start": start,
+                "segment_end": end,
                 "tokens": tokens
             }
             merged_segments.append(seg_dict)