Update README.md

README.md

@@ -51,10 +51,10 @@ processor = BridgeTowerProcessor.from_pretrained("BridgeTower/bridgetower-large-
 model = BridgeTowerForContrastiveLearning.from_pretrained("BridgeTower/bridgetower-large-itm-mlm-itc")
 
 inputs  = processor(images, texts, padding=True, return_tensors="pt")
-outputs = model(**inputs, labels=torch.ones(2))
+outputs = model(**inputs)
 
 inputs  = processor(images, texts[::-1], padding=True, return_tensors="pt")
-outputs_swapped = model(**inputs, labels=torch.ones(2))
+outputs_swapped = model(**inputs)
 
 print('Loss', outputs.loss.item())
 print('Loss with swapped images', outputs_swapped.loss.item())