Update README.md

README.md

@@ -40,14 +40,20 @@ pip install transformers torch accelerate
 
 Then the model can be downloaded and used for inference:
 ```py
-from transformers import AutoModelForSequenceClassification, AutoTokenizer
+from transformers import pipeline
 
-model = AutoModelForSequenceClassification.from_pretrained("identrics/wasper_propaganda_detection_bg", num_labels=2)
-tokenizer = AutoTokenizer.from_pretrained("identrics/wasper_propaganda_detection_bg")
+labels_map = {"LABEL_0": "No Propaganda", "LABEL_1": "Propaganda"}
 
-tokens = tokenizer("Газа евтин, американското ядрено гориво евтино, пълно с фотоволтаици а пък тока с 30% нагоре. Защо ?", return_tensors="pt")
-output = model(**tokens)
-print(output.logits)
+pipe = pipeline(
+    "text-classification",
+    model="identrics/wasper_propaganda_detection_bg",
+    tokenizer="identrics/wasper_propaganda_detection_bg",
+)
+
+text = "Газа евтин, американското ядрено гориво евтино, пълно с фотоволтаици а пък тока с 30% нагоре. Защо ?"
+
+prediction = pipe(text)
+print(labels_map[prediction[0]["label"]])
 ```