using uploaded model

app.py

@@ -4,6 +4,7 @@ import torch.nn as nn
 import torch.nn.functional as F
 import random
 import os
+import huggingface_hub
 
 if torch.cuda.is_available():
     device = torch.device("cuda:0")
@@ -46,7 +47,11 @@ class Net(nn.Module):
 net = Net()
 net.to(device)
 
-model = "1655322438.4798265_Acc0.693_modelweights.pth"
+HF_TOKEN = os.environ['HF_TOKEN']
+
+model = huggingface_hub.hf_hub_download(
+    'danyalmalik/sceneryclassifier', '1655322438.4798265_Acc0.693_modelweights.pth', use_auth_token=HF_TOKEN)
+
 net.load_state_dict(torch.load(os.path.join('/models/', model)))
 
 labels = ['Buildings', 'Forest', 'Glacier', 'Mountain', 'Sea', 'Street']

models/1655322438.4798265_Acc0.693_modelweights.pth

@@ -1,3 +0,0 @@
-version https://git-lfs.github.com/spec/v1
-oid sha256:bab05245bca6c06515fec6bd9d4f72a8d448fc23f81b888dc1797571136cefc3
-size 60034055