sinensia commited on
Commit
97bc050
1 Parent(s): 2635d47

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -3
README.md CHANGED
@@ -1,3 +1,9 @@
1
- ---
2
- license: mit
3
- ---
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ ---
4
+ Model Distilbert fine-tuned with imdb reviews for sentiment analysis
5
+
6
+ from transformers import DistilBertForSequenceClassification, DistilBertTokenizerFast
7
+
8
+ model = DistilBertForSequenceClassification.from_pretrained("./distilbert-finetuned-imdb")
9
+ tokenizer = DistilBertTokenizerFast.from_pretrained("./distilbert-finetuned-imdb")