Spaces:
Sleeping
Sleeping
Commit
•
761f68b
1
Parent(s):
6388886
mapped to cpu
Browse files
app.py
CHANGED
|
@@ -8,7 +8,7 @@ class_names = ['cat','dog']
|
|
| 8 |
|
| 9 |
|
| 10 |
model,transforms = model_classification()
|
| 11 |
-
model.load_state_dict(torch.load(path))
|
| 12 |
|
| 13 |
|
| 14 |
def predict(img):
|
|
|
|
| 8 |
|
| 9 |
|
| 10 |
model,transforms = model_classification()
|
| 11 |
+
model.load_state_dict(torch.load(path,map_location=torch.device('cpu')))
|
| 12 |
|
| 13 |
|
| 14 |
def predict(img):
|