arnabdhar commited on
Commit
805b2c4
1 Parent(s): 9fa0712

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -1
README.md CHANGED
@@ -14,6 +14,10 @@ metrics:
14
  model-index:
15
  - name: tiny-imdb
16
  results: []
 
 
 
 
17
  ---
18
 
19
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -73,4 +77,4 @@ The following hyperparameters were used during training:
73
  - Transformers 4.35.2
74
  - Pytorch 2.1.0+cu118
75
  - Datasets 2.15.0
76
- - Tokenizers 0.15.0
 
14
  model-index:
15
  - name: tiny-imdb
16
  results: []
17
+ datasets:
18
+ - imdb
19
+ library_name: transformers
20
+ pipeline_tag: text-classification
21
  ---
22
 
23
  <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 
77
  - Transformers 4.35.2
78
  - Pytorch 2.1.0+cu118
79
  - Datasets 2.15.0
80
+ - Tokenizers 0.15.0