Switch to a smaller OWL-ViT model

app.py

@@ -15,11 +15,10 @@ import tokenizers
 DEBUG = False
 if DEBUG:
     MODEL = "vit-base-patch32"
-    OWL_MODEL = f"google/owlvit-base-patch32"
 else:
     MODEL = "vit-large-patch14-336"
-    OWL_MODEL = f"google/owlvit-large-patch14"
 CLIP_MODEL = f"openai/clip-{MODEL}"
+OWL_MODEL = f"google/owlvit-base-patch32"
 
 if not DEBUG and torch.cuda.is_available():
     device = torch.device("cuda")