Update app.py

app.py

@@ -1,6 +1,5 @@
 import os
 os.system("git clone https://github.com/facebookresearch/omnivore.git")
-os.system("pwd")
 import json
 from typing import List
 
@@ -16,9 +15,10 @@ import gradio as gr
 # Device on which to run the model
 # Set to cuda to load on GPU
 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="omnivore_swinB",pretrained=True)
+model_name = "omnivore_swinB"
+model = torch.hub.load("omnivore",source='local', model=model_name,pretrained=False)
 
 # Set to eval mode and move to desired device
 model = model.to(device)
@@ -69,3 +69,4 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2201.083
 
 
 gr.Interface(inference, inputs, outputs, title=title, description=description, article=article, examples=[['library.jpg']]).launch(enable_queue=True,cache_examples=True)
+