Spaces:

Andy1621
/

uniformer_image_demo

Runtime error

Andy1621 commited on Feb 7, 2022

Commit

76085cf

1 Parent(s): 856cff1

debug hf_hub_download

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ from huggingface_hub import hf_hub_download
 # Set to cuda to load on GPU
 device = "cpu"
 # os.system("wget https://cdn-lfs.huggingface.co/Andy1621/uniformer/fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775")
-model_path = hf_hub_download(repo_id="https://huggingface.co/Andy1621/uniformer", filename="uniformer_small_in1k.pth")
 # Pick a pretrained model
 model = uniformer_small()
 # state_dict = torch.load('fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775', map_location='cpu')

 # Set to cuda to load on GPU
 device = "cpu"
 # os.system("wget https://cdn-lfs.huggingface.co/Andy1621/uniformer/fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775")
+model_path = hf_hub_download(repo_id="https://huggingface.co/Andy1621/uniformer", filename="uniformer_small_in1k.pth", revision="main")
 # Pick a pretrained model
 model = uniformer_small()
 # state_dict = torch.load('fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775', map_location='cpu')