Update README.md

README.md

@@ -8,6 +8,12 @@ metrics:
 model-index:
 - name: bert_clf_results
   results: []
+datasets:
+- app_reviews
+language:
+- en
+library_name: transformers
+pipeline_tag: text-classification
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -57,10 +63,30 @@ The following hyperparameters were used during training:
 | 0.7209        | 3.0   | 16204 | 0.8852          | 0.7121   |
 | 0.4274        | 4.0   | 21604 | 0.9611          | 0.7011   |
 
+### Code Implementation
+```
+from transformers import AutoTokenizer
+from transformers import AutoModelForSequenceClassification
+
+tokenizer = AutoTokenizer.from_pretrained("Andyrasika/bert_clf_results")
+inputs = tokenizer(prompt, return_tensors="pt")
+
+
+model = AutoModelForSequenceClassification.from_pretrained("Andyrasika/bert_clf_results")
+with torch.no_grad():
+    logits = model(**inputs).logits
+
+predicted_class_id = logits.argmax().item()
+model.config.id2label[predicted_class_id]
+```
+Output
+```
+'LABEL_4'
+```
 
 ### Framework versions
 
 - Transformers 4.36.2
 - Pytorch 2.1.0+cu118
 - Datasets 2.16.0
-- Tokenizers 0.15.0
+- Tokenizers 0.15.0