anahita-b commited on
Commit
27d6951
1 Parent(s): eb0de84

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -48,7 +48,7 @@ for text in texts:
48
  # prepare inputs
49
  encoding = processor(image, text, return_tensors="pt")
50
  outputs = model(**encoding)
51
- scores[text] = outputs.logits[0, :].item()
52
  ```
53
 
54
  Here is how to use this model to perfom masked language modeling:
@@ -74,7 +74,7 @@ outputs = model(**encoding)
74
  results = processor.decode(outputs.logits.argmax(dim=-1).squeeze(0).tolist())
75
 
76
  print(results)
77
- a cat looking out of the window.
78
  ```
79
 
80
  ### Limitations and bias
 
48
  # prepare inputs
49
  encoding = processor(image, text, return_tensors="pt")
50
  outputs = model(**encoding)
51
+ scores[text] = outputs.logits[0,1].item()
52
  ```
53
 
54
  Here is how to use this model to perfom masked language modeling:
 
74
  results = processor.decode(outputs.logits.argmax(dim=-1).squeeze(0).tolist())
75
 
76
  print(results)
77
+ .a cat looking out of the window.
78
  ```
79
 
80
  ### Limitations and bias