3oclock commited on
Commit
cef0cf1
·
verified ·
1 Parent(s): 8248e1c

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -19,7 +19,7 @@ model-index:
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
- value: 0.928
23
  ---
24
 
25
  # Model Card for DistilBERT Fine-Tuned on IMDb Dataset
@@ -83,6 +83,6 @@ Use the code below to get started with the model:
83
  ```python
84
  from transformers import pipeline
85
 
86
- classifier = pipeline("sentiment-analysis", model="your-model-id")
87
  result = classifier("I loved this movie!")
88
  print(result)
 
19
  metrics:
20
  - name: Accuracy
21
  type: accuracy
22
+ value: 0.9316
23
  ---
24
 
25
  # Model Card for DistilBERT Fine-Tuned on IMDb Dataset
 
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)