Update README.md

README.md

@@ -51,7 +51,7 @@ 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,device=device))
+outputs = model(**inputs, labels=torch.ones(2))
 
 inputs  = processor(images, texts[::-1], padding=True, return_tensors="pt")
 outputs_swapped = model(**inputs, labels=torch.ones(2))