Transformers
PyTorch
English
bridgetower
Inference Endpoints
anahita-b commited on
Commit
8786058
1 Parent(s): 5e83945

Update model name

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -40,8 +40,8 @@ from transformers import BridgeTowerProcessor, BridgeTowerForImageAndTextRetriev
40
  image_dir = "/datasets/COCO2017/val2017"
41
  search_text = "a woman holding an umbrella"
42
 
43
- processor = BridgeTowerProcessor.from_pretrained(("BridgeTower/bridgetower-base-itm"))
44
- model = BridgeTowerForImageAndTextRetrieval.from_pretrained("BridgeTower/bridgetower-base-itm")
45
 
46
  max_score = float('-inf')
47
  best_match_image = None
 
40
  image_dir = "/datasets/COCO2017/val2017"
41
  search_text = "a woman holding an umbrella"
42
 
43
+ processor = BridgeTowerProcessor.from_pretrained(("BridgeTower/bridgetower-base-itm-mlm"))
44
+ model = BridgeTowerForImageAndTextRetrieval.from_pretrained("BridgeTower/bridgetower-base-itm-mlm")
45
 
46
  max_score = float('-inf')
47
  best_match_image = None