Update app.py

app.py

@@ -29,6 +29,7 @@ from huggingface_hub import hf_hub_download
 #!pip install gradio
 import gradio as gr
 
+import shutil
 
 
 # 오디오 변환 mp4 --> wav
@@ -60,6 +61,9 @@ def seprate_speaker(audio_file, pipeline):
         # 발화 구간을 해당 화자의 리스트에 추가
         segment_waveform = waveform[:, int(start_time * sample_rate):int(end_time * sample_rate)]
         speaker_segments[speaker].append(segment_waveform)
+# 폴더가 존재하면 삭제
+    if os.path.exists("/tmp/wav" ):
+        shutil.rmtree("/tmp/wav" )  # 폴더 및 내부 파일 삭제
 
     # 각 화자별로 모든 발화 구간을 하나의 파일로 이어붙여 저장
     for speaker, segments in speaker_segments.items():
@@ -67,6 +71,8 @@ def seprate_speaker(audio_file, pipeline):
         combined_waveform = torch.cat(segments, dim=1)
         #current_path = os.getcwd()
         output_path = "/tmp/wav"    # 경로
+
+
         os.makedirs(output_path, exist_ok=True) # 경로가 없으면 생성
         output_filename = os.path.join(output_path,f"{speaker}.wav")