Update app.py

app.py

@@ -20,7 +20,7 @@ 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)
-checkpoint = torch.load("./focalnet_base_iso_16.pth")
+checkpoint = torch.load("./focalnet_base_iso_16.pth", map_location="cpu")
 model.load_state_dict(checkpoint["model"])
 model.eval()