Update README.md

README.md

@@ -5,40 +5,18 @@ tags:
 model-index:
 - name: abhirajeshbhai/symptom-2-disease-net
   results: []
+language:
+- en
 ---
 
 <!-- This model card has been generated automatically according to the information Keras had access to. You should
 probably proofread and complete it, then remove this comment. -->
 
-# abhirajeshbhai/symptom-2-disease-net
-
-This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on an unknown dataset.
-It achieves the following results on the evaluation set:
-- Train Loss: 0.0803
-- Train Accuracy: 1.0
-- Validation Loss: 0.1296
-- Validation Accuracy: 0.9583
-- Epoch: 4
-
 ## Model description
 
-More information needed
-
-## Intended uses & limitations
-
-More information needed
-
-## Training and evaluation data
-
-More information needed
-
-## Training procedure
-
-### Training hyperparameters
+This model is trained on a Symptoms to Disease [Dataset]("https://www.kaggle.com/datasets/niyarrbarman/symptom2disease"). 
+The model takes in a symptom prompt and classifies the most probable disease from it. 
 
-The following hyperparameters were used during training:
-- optimizer: {'name': 'Adam', 'weight_decay': None, 'clipnorm': None, 'global_clipnorm': None, 'clipvalue': None, 'use_ema': False, 'ema_momentum': 0.99, 'ema_overwrite_frequency': None, 'jit_compile': True, 'is_legacy_optimizer': False, 'learning_rate': 3e-05, 'beta_1': 0.9, 'beta_2': 0.999, 'epsilon': 1e-07, 'amsgrad': False}
-- training_precision: float32
 
 ### Training results
 
@@ -55,4 +33,4 @@ The following hyperparameters were used during training:
 
 - Transformers 4.28.1
 - TensorFlow 2.12.0
-- Tokenizers 0.13.3
+- Tokenizers 0.13.3