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 |
|