Update app.py

app.py

@@ -17,7 +17,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/",source='local', model="swinB_checkpoint.torch",pretrained=True)
+model = torch.hub.load("/home/user/app/omni",source='local', model="swinB_checkpoint.torch",pretrained=False)
 
 # Set to eval mode and move to desired device
 model = model.to(device)