Update README.md

README.md

@@ -21,24 +21,27 @@ It's important to note that this model has been fine-tuned solely on Arabic char
 
 # How to use
 ## Setup:
-```
+```python
 !git lfs install
 !git clone https://huggingface.co/Sifal/dzarashield
 %cd dzarashield
 
 from model import BertClassifier
 from transformers import PreTrainedTokenizerFast
+import torch
+
+MODEL_PATH = "./pytorch_model.bin"
+TOKENIZER_PATH = "./tokenizer.json"
 
 dzarashield = BertClassifier()
-PATH = "./pytorch_model.bin"
+dzarashield.load_state_dict(torch.load(MODEL_PATH))
 
-dzarashield.load_state_dict(torch.load(PATH))
-tokenizer = PreTrainedTokenizerFast(tokenizer_file="tokenizer.json")
+tokenizer = PreTrainedTokenizerFast(tokenizer_file=TOKENIZER_PATH)
 
 ```
 ## Example:
 
-```
+```python
 idx_to_label = {0: 'non-hate', 1: 'hate'}
 sentences = ['يا وحد الشموتي، تكول دجاج آآآه', 'واش خويا راك غايا؟']