Hiroshi99 commited on
Commit
dc40379
·
verified ·
1 Parent(s): 7adf460

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -35,7 +35,7 @@ class LinearRegression(nn.Module):
35
  return self.linear(x)
36
 
37
  # Charger le modèle complet
38
- model = torch.load(MODEL_PATH, map_location=torch.device('cpu'),, weights_only=False)
39
  model.eval()
40
 
41
  # Définir les noms des colonnes d'entrée
 
35
  return self.linear(x)
36
 
37
  # Charger le modèle complet
38
+ model = torch.load(MODEL_PATH, map_location=torch.device('cpu'), weights_only=False)
39
  model.eval()
40
 
41
  # Définir les noms des colonnes d'entrée