Spaces:

HaohuaLv
/

one-shot_object_detection

Running

HaohuaLv commited on Oct 25, 2023

Commit

58a00fc

•

1 Parent(s): 1bb2206

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,8 +7,8 @@ from transformers.models.owlvit.modeling_owlvit import box_iou
 from functools import partial
-processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch16")
-model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch16")
 from transformers.models.owlvit.modeling_owlvit import OwlViTImageGuidedObjectDetectionOutput

 from functools import partial
+processor = OwlViTProcessor.from_pretrained("google/owlvit-base-patch32")
+model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
 from transformers.models.owlvit.modeling_owlvit import OwlViTImageGuidedObjectDetectionOutput