LiamKhoaLe commited on
Commit
581d6f2
·
1 Parent(s): 84ec10b

Add model cache dir

Browse files
Files changed (1) hide show
  1. 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!")