Update app.py

app.py

@@ -45,9 +45,7 @@ def seprate_speaker(audio_file, pipeline):
     audio = Audio()
     waveform, sample_rate = torchaudio.load(audio_file)
     diarization = pipeline(audio_file)
-    # 폴더가 존재하면 삭제
-    if os.path.exists("/tmp/wav" ):
-        shutil.rmtree("/tmp/wav" )  # 폴더 및 내부 파일 삭제
+
         
     # 화자별로 발화 구간을 저장할 딕셔너리 초기화
     speaker_segments = {}
@@ -141,6 +139,10 @@ def real_fake_check(list_dir, path, model):
 
 
 def main(file_name):
+        # 폴더가 존재하면 삭제
+    if os.path.exists("/tmp/wav" ):
+        shutil.rmtree("/tmp/wav" )  # 폴더 및 내부 파일 삭제
+        
     hf_token = os.getenv("HUGGINGFACE_TOKEN")
     pipeline = Pipeline.from_pretrained("pyannote/speaker-diarization-3.1",use_auth_token=hf_token)
     #device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')