Update README.md

README.md

@@ -1,6 +1,9 @@
 ---
 base_model: openGPT-X/Teuken-7B-instruct-research-v0.4
 library_name: peft
+license: mit
+language:
+- de
 ---
 
 # Model Card for Model ID
@@ -41,6 +44,32 @@ library_name: peft
 
 <!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
 
+How to use the model:
+
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+tokenizer = AutoTokenizer.from_pretrained("openGPT-X/Teuken-7B-instruct-research-v0.4")
+
+model = AutoModelForSequenceClassification.from_pretrained("MB55/teuken7b-advance-classifier")
+
+model.eval()
+
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
+    if "token_type_ids" in inputs:
+        del inputs["token_type_ids"]
+    with torch.no_grad():
+        outputs = model(**inputs)
+    logits = outputs.logits
+    predicted_class = logits.argmax(dim=-1).item()
+    return predicted_class
+
+text = "Ich bin da."
+prediction = predict(text)
+
+print(f"Predicted class: {prediction}")
+
 [More Information Needed]
 
 ### Downstream Use [optional]