Nighty3912 commited on
Commit
74f559e
·
verified ·
1 Parent(s): 5aa892a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -32,6 +32,8 @@ from u2net import U2NETP
32
  # ---------------------
33
  # Create a cache folder for models
34
  # ---------------------
 
 
35
  CACHE_DIR = os.path.join(os.path.dirname(__file__), ".cache")
36
  os.makedirs(CACHE_DIR, exist_ok=True)
37
 
 
32
  # ---------------------
33
  # Create a cache folder for models
34
  # ---------------------
35
+ if os.path.exists(CACHE_DIR):
36
+ shutil.rmtree(CACHE_DIR)
37
  CACHE_DIR = os.path.join(os.path.dirname(__file__), ".cache")
38
  os.makedirs(CACHE_DIR, exist_ok=True)
39