Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -62,7 +62,7 @@ def fit_picture(img):
|
|
| 62 |
return fin_img
|
| 63 |
|
| 64 |
def load_model(filename):
|
| 65 |
-
data = torch.load(filename)
|
| 66 |
recognizer.load_state_dict(data["recognizer"])
|
| 67 |
optimizer.load_state_dict(data["optimizer"])
|
| 68 |
|
|
|
|
| 62 |
return fin_img
|
| 63 |
|
| 64 |
def load_model(filename):
|
| 65 |
+
data = torch.load(filename, map_location=torch.device('cpu'), weights_only=True)
|
| 66 |
recognizer.load_state_dict(data["recognizer"])
|
| 67 |
optimizer.load_state_dict(data["optimizer"])
|
| 68 |
|