Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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')
|
| 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
|