delete unused model weights FIXED

utils.py

@@ -118,8 +118,18 @@ def df_h():
 
 def delete_unused_bin_files(model_id: str):
     target_path = f"/data/models--poltextlab--{model_id}"
-    bin_files = glob.glob(f"{target_path}/**/*.bin", recursive=True)
-    for file_path in bin_files:
+
+    # delete .bin files in blobs/
+    blob_bins = glob.glob(f"{target_path}/blobs/**/*.bin", recursive=True)
+    
+    # delete .bin files in snapshots/, except config.json
+    snapshot_bins = glob.glob(f"{target_path}/snapshots/**/*.bin", recursive=True)
+
+    files_to_delete = blob_bins + snapshot_bins
+
+    for file_path in files_to_delete:
+        if os.path.basename(file_path) == "config.json":
+            continue
         if os.path.isfile(file_path):
             print(f"Deleting: {file_path}")
             os.remove(file_path)