Update app.py

app.py

@@ -8,11 +8,14 @@ import streamlit as st
 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 = models.efficientnet_b7(pretrained=False)
-model = torch.load("efficientnet_b7_best.pth", map_location=device)
-model.eval()  # Mode évaluation
+model = models.efficientnet_b7(pretrained=False)  # Initialize the model architecture
+model.load_state_dict(torch.load("efficientnet_b7_best.pth", map_location=device))  # Load the saved weights
+model = model.to(device)  # Move model to the appropriate device (GPU/CPU)
+model.eval()  # Set the model to evaluation mode
+
 
 # Définir la taille de l'image
 image_size = (224, 224)