Spaces:
Running
Running
Update app.py
Browse files
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 |
|