zavora commited on
Commit
a65bc24
·
verified ·
1 Parent(s): cc498d6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -1
README.md CHANGED
@@ -1,3 +1,8 @@
 
 
 
 
 
1
  # bert-sentiment-imdb
2
 
3
  This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
@@ -8,4 +13,4 @@ This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
8
  - **Usage**:
9
  ```python
10
  from transformers import pipeline
11
- classifier = pipeline("sentiment-analysis", model="zavora/bert-sentiment-imdb")
 
1
+ ---
2
+ license: mit
3
+ base_model:
4
+ - google-bert/bert-base-uncased
5
+ ---
6
  # bert-sentiment-imdb
7
 
8
  This is a fine-tuned BERT model for sentiment analysis on the IMDB dataset.
 
13
  - **Usage**:
14
  ```python
15
  from transformers import pipeline
16
+ classifier = pipeline("sentiment-analysis", model="zavora/bert-sentiment-imdb")