Update README.md

README.md

@@ -11,23 +11,20 @@ base_model:
 tags:
 - text2text-generation
 ---
-# Arabic -> Hebrew translator
+# Arabic to Hebrew Translator (Israeli Arabic Dialect)
 
-Fine-tuning [Helsinki-NLP/opus-mt-ar-he](https://huggingface.co/Helsinki-NLP/opus-mt-ar-he) for 15 epochs.
-The model is designed to translate an Israeli Arabic dialect
+This model is a fine-tuned version of [Helsinki-NLP/opus-mt-ar-he](https://huggingface.co/Helsinki-NLP/opus-mt-ar-he) on an Israeli Arabic dialect corpus, trained for 15 epochs. It aims to translate informal conversational phrases in Israeli Arabic into Hebrew, making it suitable for applications involving colloquial or everyday expressions.
 
+## Example Usage
 
-# Example usage
+Before using the fine-tuned model, load the tokenizer from the base model:
 
 ```python
-from transformers import pipeline
-trans = pipeline("translation", "HebArabNlpProject/mt-ar-he")
-trans("رحت أشتري كتاب")
-```
-```
-Out[1]: [{'translation_text': 'הלכתי לקנות ספר'}]
-```
+from transformers import pipeline, AutoTokenizer
 
-# Attribution
-Created by Matrix 
-Contact: Uri Meir
+# Load tokenizer from the base model
+tokenizer = AutoTokenizer.from_pretrained("Helsinki-NLP/opus-mt-ar-he")
+
+# Initialize the translation pipeline with the fine-tuned model and base tokenizer
+trans = pipeline("translation", model="HebArabNlpProject/mt-ar-he", tokenizer=tokenizer)
+print(trans("رحت أشتري كتاب"))