debug

app.py

@@ -11,10 +11,10 @@ import gradio as gr
 # Device on which to run the model
 # Set to cuda to load on GPU
 device = "cpu"
-os.system("wget https://huggingface.co/Andy1621/uniformer/blob/main/uniformer_small_in1k.pth")
+os.system("wget https://cdn-lfs.huggingface.co/Andy1621/uniformer/fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775")
 # Pick a pretrained model 
 model = uniformer_small()
-state_dict = torch.load('uniformer_small_in1k.pth', map_location='cpu')
+state_dict = torch.load('fd192c31f8bd77670de8f171111bd51f56fd87e6aea45043ab2edc181e1fa775', map_location='cpu')
 model.load_state_dict(state_dict['model'])
 
 # Set to eval mode and move to desired device