azherali commited on
Commit
7f471a8
·
verified ·
1 Parent(s): eae5abd

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -2
README.md CHANGED
@@ -56,9 +56,10 @@ It is designed for **domain adaptation**, helping DistilBERT better understand t
56
  ```python
57
  from transformers import pipeline
58
 
59
- nlp = pipeline("fill-mask", model="azherali/distilbert-imdb_mask_model")
60
 
61
- nlp("This movie was absolutely [MASK] and the performances were stunning.")
 
62
  # [{'sequence': 'this movie was absolutely fantastic ...', 'score': ...}, ...]
63
 
64
  for x in pipe(text):
 
56
  ```python
57
  from transformers import pipeline
58
 
59
+ pipe = pipeline("fill-mask", model="azherali/distilbert-imdb_mask_model")
60
 
61
+ text = "This movie was absolutely [MASK] and the performances were stunning."
62
+ pipe(text)
63
  # [{'sequence': 'this movie was absolutely fantastic ...', 'score': ...}, ...]
64
 
65
  for x in pipe(text):