Update weights.py

weights.py

@@ -1,6 +1,5 @@
 import os
 import gdown
-import torch
 from pathlib import Path
 
 def download_weights():
@@ -8,10 +7,10 @@ def download_weights():
     Path("weights/icon_detect").mkdir(parents=True, exist_ok=True)
     Path("weights/icon_caption_florence").mkdir(parents=True, exist_ok=True)
     
-    # Google Drive file IDs and their corresponding local paths
+    # Correct file mapping with Drive file IDs and their corresponding local paths
     files_to_download = {
-        "https://drive.google.com/file/d/1hUCqZ3X8mcM-KcwWFjcsFg7PA0hUvE3k": "weights/icon_caption_florence/model.safetensors",
-        "https://drive.google.com/file/d/1p-Y7rd0FfjNnv_jewCi7ZjXH3T-qtyAa": "weights/icon_detect/best.pt"
+        "1hUCqZ3X8mcM-KcwWFjcsFg7PA0hUvE3k": "weights/icon_caption_florence/model.safetensors",
+        "1p-Y7rd0FfjNnv_jewCi7ZjXH3T-qtyAa": "weights/icon_detect/best.pt"
     }
     
     for file_id, output_path in files_to_download.items():
@@ -24,4 +23,4 @@ def download_weights():
             print(f"File {output_path} already exists, skipping download")
 
 if __name__ == "__main__":
-    download_weights()
+    download_weights()