Update app.py

app.py

@@ -27,7 +27,8 @@ class MNISTNet(nn.Module):
         return x
 
 model = MNISTNet()
-model.load_state_dict(torch.hub.load_state_dict_from_url('https://huggingface.co/Gaimundo/mnist-nn/resolve/main/mnist_cnn.pt', map_location='cpu'))
+model_file = hf_hub_download(repo_id="Gaimundo/mnist-nn", filename="mnist_cnn.pt")
+model.load_state_dict(torch.load(model_file, map_location="cpu"))
 model.eval()
 
 transform = transforms.Compose([