3oclock commited on
Commit
50c7475
·
verified ·
1 Parent(s): cef0cf1

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -81,8 +81,9 @@ Users (both direct and downstream) should be made aware of the risks, biases, an
81
  Use the code below to get started with the model:
82
 
83
  ```python
84
- from transformers import pipeline
85
 
86
- classifier = pipeline("sentiment-analysis", model="3oclock/distilbert-imdb")
87
- result = classifier("I loved this movie!")
 
88
  print(result)
 
81
  Use the code below to get started with the model:
82
 
83
  ```python
84
+ from transformers import pipeline,DistilBertTokenizer
85
 
86
+ tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
87
+ classifier = pipeline("sentiment-analysis", model="3oclock/distilbert-imdb", tokenizer=tokenizer)
88
+ result = classifier("I love this movie!")
89
  print(result)