Spaces:
Running
Running
Commit
·
581d6f2
1
Parent(s):
84ec10b
Add model cache dir
Browse files- download_model.py +12 -0
download_model.py
CHANGED
|
@@ -18,3 +18,15 @@ if os.path.exists(snapshot_folders):
|
|
| 18 |
shutil.move(os.path.join(snapshot_dir, filename), MODEL_CACHE_DIR) # Move files to /app/model_cache
|
| 19 |
# Complete
|
| 20 |
print(f"✅ Model downloaded and extracted to {MODEL_CACHE_DIR}")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 18 |
shutil.move(os.path.join(snapshot_dir, filename), MODEL_CACHE_DIR) # Move files to /app/model_cache
|
| 19 |
# Complete
|
| 20 |
print(f"✅ Model downloaded and extracted to {MODEL_CACHE_DIR}")
|
| 21 |
+
|
| 22 |
+
# Checking model download and caching directory
|
| 23 |
+
print("\n📂 Checking model directory structure...")
|
| 24 |
+
|
| 25 |
+
# Find files
|
| 26 |
+
if os.path.exists(MODEL_CACHE_DIR):
|
| 27 |
+
for root, dirs, files in os.walk(MODEL_CACHE_DIR):
|
| 28 |
+
print(f"📁 {root}/")
|
| 29 |
+
for file in files:
|
| 30 |
+
print(f" 📄 {file}")
|
| 31 |
+
else:
|
| 32 |
+
print("❌ Model cache directory does not exist!")
|