Update README.md

README.md

@@ -13,4 +13,39 @@ tags: []
 
 ### Model Description
 
-### How to use the model
+### How to use the model
+
+```python 
+import pandas as pd
+import numpy as np
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+
+# Load model
+model = AutoModelForSequenceClassification.from_pretrained("lkonle/EMO_Joy_gbert")
+
+# Load tokenizer
+tokenizer = AutoTokenizer.from_pretrained("lkonle/EMO_Joy_gbert")
+tokenizer.pad_token = "[PAD]"
+tokenizer.add_special_tokens({'pad_token': '[PAD]'})
+
+# define input text
+myinput = ["Paul war sehr sehr glücklich über seinen Welpen.",
+           "Paul war sehr traurig über sein Frühstück.",
+           "Paul hatte große Langeweile."]
+
+# tokenize, encode, format as batch and return pytorch tensors
+input_ids = tokenizer.batch_encode_plus(myinput, truncation=True, padding="max_length", padding_side="right", return_tensors="pt")
+
+# predict
+logits = model(**input_ids)["logits"]
+
+# get the predicted label
+result = logits.detach().numpy()
+prediction = np.argmax(result, axis=1)
+
+# store result in pandas
+output = pd.DataFrame()
+output["inputs"] =  myinput
+output["prediction"] = prediction
+print(output)
+```