brightlembo commited on
Commit
559a45a
·
verified ·
1 Parent(s): 3e0991a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ with open("class_names.json", "r") as f:
10
 
11
  # Charger le modèle
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
- model = torch.load("efficientnet_b7_best.pth", map_location=device)
14
  model.eval() # Mode évaluation
15
 
16
  # Définir la taille de l'image
 
10
 
11
  # Charger le modèle
12
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
13
+ model = torch.load("efficientnet_b7_best.pth", map_location=device, weights_only=False )
14
  model.eval() # Mode évaluation
15
 
16
  # Définir la taille de l'image