Update app.py

app.py

@@ -1,3 +1,5 @@
+import os
+os.system("git clone https://github.com/facebookresearch/omnivore.git")
 import json
 from typing import List
 
@@ -14,7 +16,7 @@ device = "cpu"
 
 # Pick a pretrained model 
 model_name = "omnivore_swinB"
-model = torch.hub.load("facebookresearch/omnivore:main", model=model_name)
+model = torch.hub.load("omnivore", model=model_name)
 
 # Set to eval mode and move to desired device
 model = model.to(device)