one-shot_object_detection

Sleeping

HaohuaLv commited on Oct 25, 2023

Commit

1bb2206

1 Parent(s): 293d766

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-patch32")
-model = OwlViTForObjectDetection.from_pretrained("google/owlvit-base-patch32")
 from transformers.models.owlvit.modeling_owlvit import OwlViTImageGuidedObjectDetectionOutput

 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