Update app.py

app.py

@@ -31,7 +31,7 @@ with open(CLASS_NAMES_PATH, "r") as f:
 # Charger le modèle
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 try:
-    model = torch.load(MODEL_PATH, map_location=device)
+    model = torch.load(MODEL_PATH, map_location=device,weights_only=True)
     model.eval()
 except Exception as e:
     st.error(f"Erreur lors du chargement du modèle : {e}")