BridgeTower
/

bridgetower-base-itm-mlm

Inference Endpoints

Model card Files Files and versions Community

anahita-b commited on Dec 12, 2022

Commit

8786058

•

1 Parent(s): 5e83945

Update model name

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -40,8 +40,8 @@ from transformers import BridgeTowerProcessor, BridgeTowerForImageAndTextRetriev
 image_dir = "/datasets/COCO2017/val2017"
 search_text = "a woman holding an umbrella"
-processor = BridgeTowerProcessor.from_pretrained(("BridgeTower/bridgetower-base-itm"))
-model = BridgeTowerForImageAndTextRetrieval.from_pretrained("BridgeTower/bridgetower-base-itm")
 max_score = float('-inf')
 best_match_image = None

 image_dir = "/datasets/COCO2017/val2017"
 search_text = "a woman holding an umbrella"
+processor = BridgeTowerProcessor.from_pretrained(("BridgeTower/bridgetower-base-itm-mlm"))
+model = BridgeTowerForImageAndTextRetrieval.from_pretrained("BridgeTower/bridgetower-base-itm-mlm")
 max_score = float('-inf')
 best_match_image = None