Update README.md

README.md

@@ -20,7 +20,7 @@ Dzarashield is a fine-tuned model based on [DzaraBert](https://huggingface.co/Si
 It's important to note that this model has been fine-tuned solely on Arabic characters, which means that tokens from other languages have been pruned.
 
 # How to use
-
+## Setup:
 ```
 !git lfs install
 !git clone https://huggingface.co/Sifal/dzarashield
@@ -36,7 +36,22 @@ dzarashield.load_state_dict(torch.load(PATH))
 tokenizer = PreTrainedTokenizerFast(tokenizer_file="tokenizer.json")
 
 ```
+## Example:
 
+```
+idx_to_label = {0: 'non-hate', 1: 'hate'}
+sentences = ['يا وحد الشموتي، تكول دجاج آآآه', 'واش خويا راك غايا؟']
+
+def predict_label(sentence):
+    tokenized = tokenizer(sentence, return_tensors='pt')
+    with torch.no_grad():
+        outputs = dzarashield(**tokenized)
+        return idx_to_label[outputs.logits.argmax().item()]
+
+for sentence in sentences:
+    label = predict_label(sentence)
+    print(f'sentence: {sentence} label: {label}')
+```
 ## Acknowledgments
 
 Dzarashield is built upon the foundations of [Dziribert](https://huggingface.co/alger-ia/dziribert), and I am grateful for their work in making this project possible.