Upload app.py

app.py

@@ -1,3 +1,4 @@
+import gdown
 import os
 import torch
 import requests
@@ -14,34 +15,13 @@ import torchvision.transforms as transforms
 from pim_module import PluginMoodel  # Assure-toi que ce fichier est présent
 
 # === Téléchargement automatique depuis Google Drive ===
-
-def download_file_from_google_drive(file_id, destination):
-    URL = "https://drive.google.com/uc?export=download"
-    session = requests.Session()
-    response = session.get(URL, params={"id": file_id}, stream=True)
-    token = get_confirm_token(response)
-    if token:
-        params = {"id": file_id, "confirm": token}
-        response = session.get(URL, params=params, stream=True)
-    save_response_content(response, destination)
-
-def get_confirm_token(response):
-    for key, value in response.cookies.items():
-        if key.startswith("download_warning"):
-            return value
-    return None
-
-def save_response_content(response, destination):
-    CHUNK_SIZE = 32768
-    with open(destination, "wb") as f:
-        for chunk in response.iter_content(CHUNK_SIZE):
-            if chunk:
-                f.write(chunk)
-
 if not os.path.exists("weights.pt"):
-    print("Téléchargement des poids depuis Google Drive...")
+    print("Téléchargement des poids depuis Google Drive avec gdown...")
     file_id = "1Ck9qyjs4_c_fqgaEpZ0eN9jIV5TiqkXp"
-    download_file_from_google_drive(file_id, "weights.pt")
+    url = f"https://drive.google.com/uc?id={file_id}"
+    gdown.download(url, "weights.pt", quiet=False)
+
+
 
 # === Classes
 classes_list = [