Update app.py

app.py

@@ -34,7 +34,7 @@ def inference(img):
 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cpu"
-model_path = hf_hub_download(repo_id="Andy1621/uniformer", filename="uniformer_small_in1k.pth")
+model_path = hf_hub_download(repo_id="Sense-X/uniformer_image", filename="uniformer_small_in1k.pth")
 # Pick a pretrained model 
 model = uniformer_small()
 state_dict = torch.load(model_path, map_location='cpu')