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

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -13
README.md CHANGED
@@ -21,6 +21,17 @@ model-index:
21
  type: accuracy
22
  value: 0.9316
23
  ---
 
 
 
 
 
 
 
 
 
 
 
24
 
25
  # Model Card for DistilBERT Fine-Tuned on IMDb Dataset
26
 
@@ -74,16 +85,4 @@ The model was trained on a dataset of movie reviews, so it may not perform as we
74
 
75
  ### Recommendations
76
 
77
- Users (both direct and downstream) should be made aware of the risks, biases, and limitations of the model.
78
-
79
- ## How to Get Started with the Model
80
-
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)
 
21
  type: accuracy
22
  value: 0.9316
23
  ---
24
+ ## How to Get Started with the Model
25
+
26
+ Use the code below to get started with the model:
27
+
28
+ ```python
29
+ from transformers import pipeline,DistilBertTokenizer
30
+
31
+ tokenizer = DistilBertTokenizer.from_pretrained("distilbert-base-uncased")
32
+ classifier = pipeline("sentiment-analysis", model="3oclock/distilbert-imdb", tokenizer=tokenizer)
33
+ result = classifier("I love this movie!")
34
+ print(result)
35
 
36
  # Model Card for DistilBERT Fine-Tuned on IMDb Dataset
37
 
 
85
 
86
  ### Recommendations
87
 
88
+ Users (both direct and downstream) should be made aware of the risks, biases, and limitations of the model.