Nick088 commited on
Commit
bdeb650
·
verified ·
1 Parent(s): f701299

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -6
app.py CHANGED
@@ -62,12 +62,7 @@ def inference_image(image, size):
62
  def inference_video(video, size):
63
  _id = randint(1, 10000)
64
  INPUT_DIR = "tmp"
65
- # Check if the directory exists, if so remove it
66
- if os.path.exists(INPUT_DIR):
67
- shutil.rmtree(INPUT_DIR)
68
- else:
69
- # Create the directory, equivalent to 'mkdir -p'
70
- os.makedirs(INPUT_DIR, exist_ok=True)
71
  os.chdir(INPUT_DIR)
72
 
73
  upload_folder = 'upload'
 
62
  def inference_video(video, size):
63
  _id = randint(1, 10000)
64
  INPUT_DIR = "tmp"
65
+ os.makedirs(INPUT_DIR, exist_ok=True)
 
 
 
 
 
66
  os.chdir(INPUT_DIR)
67
 
68
  upload_folder = 'upload'