Update app.py

app.py

@@ -4,13 +4,14 @@ from PIL import Image
 import streamlit as st
 import json
 
+
 # Charger les noms des classes
 with open("class_names.json", "r") as f:
     class_names = json.load(f)
 
 # Charger le modèle
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = torch.load("efficientnet_b7_best.pth", map_location=device, weights_only=False )
+model = torch.load("efficientnet_b7_best.pth", map_location=device)
 model.eval()  # Mode évaluation
 
 # Définir la taille de l'image