Update app.py

app.py

@@ -62,8 +62,12 @@ def inference_image(image, size):
 def inference_video(video, size):
     _id = randint(1, 10000)
     INPUT_DIR = "tmp/" + str(_id)
-    os.system("rm -rf " + INPUT_DIR)
-    os.system("mkdir -p " + INPUT_DIR)
+    # Check if the directory exists, if so remove it
+    if os.path.exists(INPUT_DIR):
+        shutil.rmtree(INPUT_DIR)
+    else:
+        # Create the directory, equivalent to 'mkdir -p'
+        os.makedirs(INPUT_DIR, exist_ok=True)
     os.system(f"cd {INPUT_DIR}")
     upload_folder = 'upload'
     result_folder = 'results'