Update app.py
Browse files
app.py
CHANGED
|
@@ -21,7 +21,7 @@ from u2net import U2NET # Assume you copied the model code as u2net.py
|
|
| 21 |
|
| 22 |
# Load model
|
| 23 |
model = U2NET(3, 1)
|
| 24 |
-
model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu')))
|
| 25 |
model.eval()
|
| 26 |
|
| 27 |
# Preprocessing
|
|
|
|
| 21 |
|
| 22 |
# Load model
|
| 23 |
model = U2NET(3, 1)
|
| 24 |
+
model.load_state_dict(torch.load(model_path, map_location=torch.device('cpu'), weights_only=False))
|
| 25 |
model.eval()
|
| 26 |
|
| 27 |
# Preprocessing
|