Spaces:
Runtime error
Runtime error
Commit ·
2e9bd2b
1
Parent(s): a5b9f6a
made token secret
Browse files
app.py
CHANGED
|
@@ -47,8 +47,10 @@ class Net(nn.Module):
|
|
| 47 |
net = Net()
|
| 48 |
net.to(device)
|
| 49 |
|
|
|
|
|
|
|
| 50 |
model = huggingface_hub.cached_download(huggingface_hub.hf_hub_url(
|
| 51 |
-
'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth'), use_auth_token=
|
| 52 |
|
| 53 |
net.load_state_dict(torch.load(model, map_location=device))
|
| 54 |
|
|
|
|
| 47 |
net = Net()
|
| 48 |
net.to(device)
|
| 49 |
|
| 50 |
+
HF_Token = os.environ['HF_Token']
|
| 51 |
+
|
| 52 |
model = huggingface_hub.cached_download(huggingface_hub.hf_hub_url(
|
| 53 |
+
'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth'), use_auth_token=HF_Token)
|
| 54 |
|
| 55 |
net.load_state_dict(torch.load(model, map_location=device))
|
| 56 |
|