Update README.md

README.md

@@ -34,7 +34,7 @@ interests you.
 
 Here is how to use this model to get the features of a given text in PyTorch:
 ```python
-from transformers import BridgeTowerProcessor, BridgeTowerForModel
+from transformers import BridgeTowerProcessor, BridgeTowerModel
 import requests
 from PIL import Image
 
@@ -43,7 +43,7 @@ image = Image.open(requests.get(url, stream=True).raw)
 text = "hello world"
 
 processor = BridgeTowerProcessor.from_pretrained("BridgeTower/bridgetower-base")
-model = BridgeTowerForModel.from_pretrained("BridgeTower/bridgetower-base")
+model = BridgeTowerModel.from_pretrained("BridgeTower/bridgetower-base")
 # Prepare inputs
 encoding = processor(image, text, return_tensors="pt")
 # Forward pass