Update app.py

app.py

@@ -7,6 +7,7 @@ import random
 from PIL import Image
 from transparent_background import Remover
 import subprocess
+import os
 
 @spaces.GPU()
 def doo(video, mode):
@@ -21,9 +22,9 @@ def doo(video, mode):
     height = int(cap.get(cv2.CAP_PROP_FRAME_HEIGHT))
     
     tmpname = random.randint(111111111, 999999999)
-    tmp_video = f"{tmpname}_video.mp4"
-    tmp_audio = f"{tmpname}_audio.aac"
-    output_video = f"{tmpname}_output.mp4"
+    tmp_video = f"/tmp/{tmpname}_video.mp4"
+    tmp_audio = f"/tmp/{tmpname}_audio.aac"
+    output_video = f"/tmp/{tmpname}_output.mp4"
     
     writer = cv2.VideoWriter(tmp_video, cv2.VideoWriter_fourcc(*'mp4v'), fps, (width, height))
     
@@ -58,12 +59,12 @@ def doo(video, mode):
     subprocess.call(['ffmpeg', '-i', tmp_video, '-i', tmp_audio, '-c', 'copy', output_video])
     
     # 임시 파일 삭제
-    subprocess.call(['rm', tmp_video, tmp_audio])
+    os.remove(tmp_video)
+    os.remove(tmp_audio)
     
     return output_video
 
 # Gradio 인터페이스 정의
-examples = [['./input.mp4'],]
 css = """footer { visibility: hidden; }"""
 
 b_interface = gr.Interface(
@@ -73,8 +74,11 @@ b_interface = gr.Interface(
         gr.Radio(['Normal', 'Fast'], label='Select mode', value='Normal', info='Normal is more accurate, but takes longer. | Fast has lower accuracy so the process will be faster.')
     ],
     outputs="video",
-    examples=examples,
     css=css,
     title="Video Background Removal",
     description="Select a video and mode to remove the background."
-)
+)
+
+# Gradio 앱 실행
+if __name__ == "__main__":
+    b_interface.launch()