Spaces:

camenduru
/

sam2-remove-video-background

Runtime error

camenduru commited on Sep 17

Commit

8c65d88

•

1 Parent(s): c793436

Update worker_runpod.py

Files changed (1) hide show

worker_runpod.py CHANGED Viewed

@@ -180,8 +180,8 @@ def generate(input):
         if os.path.exists(result):
             os.remove(result)
         if os.path.exists(output_frames_dir):
-            os.rmdir(output_frames_dir)
         if os.path.exists(frames_dir):
-            os.rmdir(frames_dir)
 runpod.serverless.start({"handler": generate})

         if os.path.exists(result):
             os.remove(result)
         if os.path.exists(output_frames_dir):
+            shutil.rmtree(output_frames_dir)
         if os.path.exists(frames_dir):
+            shutil.rmtree(frames_dir)
 runpod.serverless.start({"handler": generate})