DavidNgoue commited on
Commit
76dc6f7
·
verified ·
1 Parent(s): 4e20d83

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -129,7 +129,7 @@ st.title("🎭 Détecteur d'Émotions en Temps Réel")
129
  def load_model():
130
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
131
  model = EmotionCNN().to(device)
132
- model.load_state_dict(torch.load("modele/cnn_emotion_model.pth", map_location=device))
133
  model.eval()
134
  return model, device
135
 
 
129
  def load_model():
130
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
131
  model = EmotionCNN().to(device)
132
+ model.load_state_dict(torch.load("cnn_emotion_model.pth", map_location=device))
133
  model.eval()
134
  return model, device
135