Update README.md

README.md

@@ -76,11 +76,11 @@ query_full = query_instruction + query
 
 # Embed image documents
 with torch.no_grad():
-    p_reps = model(text=['', '', ''], image=[image_1, image_2, image_3], tokenizer=tokenizer)
+    p_reps = model(text=['', '', ''], image=[image_1, image_2, image_3], tokenizer=tokenizer).reps
 
 # Embed text queries
 with torch.no_grad():
-    q_reps = model(text=[query_full], image=[None], tokenizer=tokenizer) # [B, s, d]
+    q_reps = model(text=[query_full], image=[None], tokenizer=tokenizer).reps # [B, s, d]
 
 # Calculate similarities
 scores = torch.matmul(q_reps, p_reps.T)