Update app.py

app.py

@@ -18,8 +18,9 @@ labels = response.text.split("\n")
 build model
 '''
 model = FocalNet(depths=[12], patch_size=16, embed_dim=768, focal_levels=[3], use_layerscale=True, use_postln=True)
-url = 'https://projects4jw.blob.core.windows.net/focalnet/release/classification/focalnet_base_iso_16.pth'
-checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", check_hash=True)
+# url = 'https://projects4jw.blob.core.windows.net/focalnet/release/classification/focalnet_base_iso_16.pth'
+# checkpoint = torch.hub.load_state_dict_from_url(url=url, map_location="cpu", check_hash=True)
+checkpoint = torch.load("./focalnet_base_iso_16.pth")
 model.load_state_dict(checkpoint["model"])
 model.eval()