Update app.py

app.py

@@ -18,7 +18,7 @@ import gradio as gr
 device = "cpu"
 # Pick a pretrained model 
 os.system("wget https://huggingface.co/akhaliq/Omnivore/resolve/main/swinB_checkpoint.torch")
-model = torch.hub.load("/home/user/app/omnivore/",source='local', model="swinB_checkpoint.torch",pretrained=True)
+model = torch.hub.load("/home/user/app/omnivore/",source='local', model="omnivore_swinB",pretrained=True)
 
 # Set to eval mode and move to desired device
 model = model.to(device)