Spaces:
Runtime error
Runtime error
Commit ·
a7394d1
1
Parent(s): b727665
fixed typos
Browse files
app.py
CHANGED
|
@@ -49,10 +49,10 @@ class Net(nn.Module):
|
|
| 49 |
net = Net()
|
| 50 |
net.to(device)
|
| 51 |
|
| 52 |
-
model = huggingface_hub.
|
| 53 |
-
'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth')
|
| 54 |
|
| 55 |
-
net.load_state_dict(torch.load(
|
| 56 |
|
| 57 |
labels = ['Buildings', 'Forest', 'Glacier', 'Mountain', 'Sea', 'Street']
|
| 58 |
|
|
|
|
| 49 |
net = Net()
|
| 50 |
net.to(device)
|
| 51 |
|
| 52 |
+
model = huggingface_hub.cached_download(huggingface_hub.hf_hub_url(
|
| 53 |
+
'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth'))
|
| 54 |
|
| 55 |
+
net.load_state_dict(torch.load(model))
|
| 56 |
|
| 57 |
labels = ['Buildings', 'Forest', 'Glacier', 'Mountain', 'Sea', 'Street']
|
| 58 |
|