Update app.py

app.py

@@ -1,5 +1,5 @@
 import torch
-from torchvision import transforms
+from torchvision import transforms, models
 from PIL import Image
 import json
 import streamlit as st
@@ -10,6 +10,7 @@ with open("class_names.json", "r") as 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