Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -93,8 +93,9 @@ def image_mod(image):
|
|
| 93 |
|
| 94 |
device = get_default_device()
|
| 95 |
PATH = "./model/model.zip"
|
|
|
|
| 96 |
model = DogBreedPretrainedWideResnet()
|
| 97 |
-
model.load_state_dict(torch.load(PATH))
|
| 98 |
breeds=['Chihuahua',
|
| 99 |
'Japanese spaniel',
|
| 100 |
'Maltese dog',
|
|
|
|
| 93 |
|
| 94 |
device = get_default_device()
|
| 95 |
PATH = "./model/model.zip"
|
| 96 |
+
map_location=torch.device('cpu')
|
| 97 |
model = DogBreedPretrainedWideResnet()
|
| 98 |
+
model.load_state_dict(torch.load(PATH,map_location))
|
| 99 |
breeds=['Chihuahua',
|
| 100 |
'Japanese spaniel',
|
| 101 |
'Maltese dog',
|