ninaa510
/

distilbert-finetuned-medical-diagnosis

Text Classification

generated_from_keras_callback

Inference Endpoints

Model card Files Files and versions Community

ninaa510 commited on May 28

Commit

aaf6b50

•

1 Parent(s): b9df3b6

Update README.md

Files changed (1) hide show

README.md +3 -1

README.md CHANGED Viewed

@@ -6,6 +6,8 @@ base_model: distilbert/distilbert-base-cased
 model-index:
 - name: distilbert-finetuned-medical-diagnosis
   results: []
 ---
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
@@ -46,4 +48,4 @@ The following hyperparameters were used during training:
 - Transformers 4.41.0
 - TensorFlow 2.15.0
 - Datasets 2.19.1
-- Tokenizers 0.19.1

 model-index:
 - name: distilbert-finetuned-medical-diagnosis
   results: []
+datasets:
+- ninaa510/diagnosis-text
 ---
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
 - Transformers 4.41.0
 - TensorFlow 2.15.0
 - Datasets 2.19.1
+- Tokenizers 0.19.1