Spaces:
Runtime error
Runtime error
edited model dict
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ to_device(model, device);
|
|
| 104 |
|
| 105 |
|
| 106 |
# loading the model
|
| 107 |
-
ckp_path = '
|
| 108 |
model.load_state_dict(torch.load(ckp_path))
|
| 109 |
model.eval()
|
| 110 |
|
|
|
|
| 104 |
|
| 105 |
|
| 106 |
# loading the model
|
| 107 |
+
ckp_path = 'indianFood-resnet50.pth'
|
| 108 |
model.load_state_dict(torch.load(ckp_path))
|
| 109 |
model.eval()
|
| 110 |
|