Update sv.py

sv.py

@@ -365,11 +365,18 @@ if __name__ == "__main__":
     audio_path = "example/mtr.mp3"  # Replace with your audio file path
     language = "yue"  # Set language to Cantonese
 
-    result = process_audio(audio_path, language)
+    # Option to run only diarization
+    diarization_only = False  # Set this to True if you want only diarization
 
-    # Save the result to mtr.txt
-    output_path = "mtr.txt"
-    with open(output_path, "w", encoding="utf-8") as f:
-        f.write(result)
+    if diarization_only:
+        diarization_segments = generate_diarization(audio_path)
+        # You can add code here to save or process the diarization results as needed
+    else:
+        result = process_audio(audio_path, language)
+
+        # Save the result to mtr.txt
+        output_path = "mtr.txt"
+        with open(output_path, "w", encoding="utf-8") as f:
+            f.write(result)
 
-    print(f"Diarization and transcription result has been saved to {output_path}")
+        print(f"Diarization and transcription result has been saved to {output_path}")