fix

app.py

@@ -1,4 +1,3 @@
-import argparse
 
 import torch
 from scipy.io.wavfile import write
@@ -35,6 +34,5 @@ app = gr.Interface(
 
 )
 
-
 app.launch(debug=True, enable_queue=True,
-)
+           )

main_pipeline.py

@@ -1,19 +1,8 @@
 import argparse
-import librosa
-import torch
-import os
-
-from scipy.io.wavfile import write
-from tqdm import tqdm
 
 from utils.denoise_pipeline import denoise
 from utils.diarization_pipeline import DiarizationPipeline
 
-import numpy as np
-
-import pandas as pd
-import soundfile as sf
-
 
 class CleaningPipeline:
     def __init__(self, device):
@@ -48,5 +37,3 @@ if __name__ == '__main__':
     opt = parser.parse_args()
     cleaning_pipeline = CleaningPipeline('cuda:0')
     cleaning_pipeline(input_audio_path=opt.audio_path)
-    # for _ in tqdm(range(10)):
-    #     main_pipeline(audio_path=opt.audio_path, device=opt.device)

utils/diarization_pipeline.py

@@ -648,7 +648,7 @@ class DiarizationPipeline:
         labels = durs[durs['duration'] / durs.sum()['duration'] > 0.015].index
         return segments[segments.label.isin(labels)]
 
-    def __call__(self, input_wav_path: str)-> dict:
+    def __call__(self, input_wav_path: str) -> dict:
 
         segments = self.diar.diarize(input_wav_path,
                                      num_speakers=None,