Update app.py

app.py

@@ -66,8 +66,8 @@ def seprate_speaker(audio_file, pipeline):
     for speaker, segments in speaker_segments.items():
         # 화자의 모든 발화 구간을 이어붙임
         combined_waveform = torch.cat(segments, dim=1)
-        current_path = os.getcwd()
-        output_path = current_path+"/wav"    # 경로
+        #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")
 
@@ -138,8 +138,8 @@ def main(file_name):
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     video_file = file_name #deepfake #meganfox.mp4'
-    current_path = os.getcwd()
-    audio_file = current_path+'/output_audio.wav'  # 저장할 오디오 파일의 경로, 이름 지정
+    #current_path = os.getcwd()
+    audio_file = '/tmp/output_audio.wav'  # 저장할 오디오 파일의 경로, 이름 지정
 
     extract_audio_from_video(video_file, audio_file)
 
@@ -164,8 +164,8 @@ def main(file_name):
     #real_path = '/content/drive/MyDrive/Celeb-DF-v2/Celeb-real'
 
     #real = os.listdir(real_path)
-    current_path = os.getcwd()
-    fake_path = current_path+'/wav'
+    #current_path = os.getcwd()
+    fake_path = '/tmp/wav'
     fake = os.listdir(fake_path)
 
     rf_check = real_fake_check(fake, fake_path,model)       #fake dataset\