Update README.md

README.md

@@ -11,17 +11,20 @@ tags:
 # Model Card for identrics/BG_propaganda_detector
 
 
-## Model Description
-
-<!-- Provide a longer summary of what this model is. -->
 
 
+## Model Description
 
 - **Developed by:** Identrics
 - **Language:** Bulgarian
 - **License:** apache-2.0
 - **Finetuned from model:** INSAIT-Institute/BgGPT-7B-Instruct-v0.2
+- **Context window :** 8192 tokens
+
+## Model Description
 
+This model consists of a fine-tuned version of BgGPT-7B-Instruct-v0.2 for a propaganda detection task. It is effectively a binary classifier, determining wether propaganda is present in the output string.
+This model was created by [`Identrics`](https://identrics.ai/), in the scope of the Wasper project.
 
 
 ## Uses
@@ -30,7 +33,22 @@ To be used as a binary classifier to identify if propaganda is present in a stri
 
 ### Example
 
-<!-- This section is for the model use without fine-tuning or plugging into a larger ecosystem/app. -->
+First install direct dependencies:
+```
+pip install transformers torch accelerate
+```
+
+Then the model can be downloaded and used for inference:
+```py
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+
+model = AutoModelForSequenceClassification.from_pretrained("identrics/EN_propaganda_detector", num_labels=2)
+tokenizer = AutoTokenizer.from_pretrained("identrics/EN_propaganda_detector")
+
+tokens = tokenizer("Our country is the most powerful country in the world!", return_tensors="pt")
+output = model(**tokens)
+print(output.logits)
+```
 
 
 ## Training Details