Update README.md

README.md

@@ -63,4 +63,80 @@ By Language:
 
 ![](confusion_matrix_pt.svg)
 
-![](confusion_matrix_es.svg)
+![](confusion_matrix_es.svg)
+
+
+
+## Usage example
+
+```python
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("LenDigLearn/formality-classifier-mdeberta-v3-base")
+model = AutoModelForSequenceClassification.from_pretrained('LenDigLearn/formality-classifier-mdeberta-v3-base')
+
+def get_result(text):
+    inputs = tokenizer(text, return_tensors="pt")
+    with torch.no_grad():
+        logits = model(**inputs).logits
+    predicted_class_id = logits.argmax().item()
+    return model.config.id2label[predicted_class_id]
+
+
+print("DE:")
+texts_de = [
+    "Verschwinde", "Nein", "Ja", "vielleicht", "Warum bist du so?",
+    "Können Sie mir spontan dabei helfen?", "Bitte senden Sie uns die nötigen Unterlagen zu.", "Dies müssen Sie selbst entscheiden, wenn Sie den entsprechenden Punkt erreicht haben.", "Sie sind also Herr Müller.", "Bitte helfen Sie mir!",
+    "Man muss schon wissen, was dann passiert.", "Als nächstes kommen 4g Champignons und 500g Mehl dazu.", "Bananen sind krumm.", "Das ist eine Tatsache, die unumstößlich ist.", "Hilfestellungen sind unter \"Hilfe\" zu finden."
+]
+for text in texts_de:
+    print(get_result(text))
+
+print("-----------\nEN:")
+texts_en = [
+    "Piss off", "No", "Yes", "maybe", "Why are you like this?",
+    "Could you help me spontaneously?", "Please send me the necessary documents.", "You will have to decide this individually as soon as you have reached the relevant point.", "I presume you are Mr. Müller?", "Please offer me your support!",
+    "One would have to know what happens then.", "Then, we add 4g Mushrooms and 500g flour.", "Bananas are usually curved.", "That is an irrefutable fact.", "You can find helpful tutorials under \"help\"."
+]
+for text in texts_en:
+    print(get_result(text))
+```
+
+**Outputs:**
+
+```bash
+DE:
+informal
+informal
+informal
+informal
+informal
+formal
+formal
+formal
+formal
+formal
+neutral
+neutral
+neutral
+neutral
+neutral
+-----------
+EN:
+informal
+informal
+informal
+informal
+informal
+formal
+formal
+formal
+formal
+formal
+neutral
+neutral
+neutral
+neutral
+neutral
+```