Update README.md

README.md

@@ -63,18 +63,21 @@ This model is designed for translating English content into colloquial Egyptian
 Use the code below to get started with the model:
 
 ```python
-from transformers import MarianMTModel, MarianTokenizer
-
-model_name = "Omar-youssef/english-egyptian-arabic-translator"
-tokenizer = MarianTokenizer.from_pretrained(model_name)
-model = MarianMTModel.from_pretrained(model_name)
-
-# Example usage
-text = "Hello, how are you today?"
-inputs = tokenizer(text, return_tensors="pt")
-translated = model.generate(**inputs)
-output = tokenizer.decode(translated[0], skip_special_tokens=True)
-print(output)
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("Omar-youssef/english-egyptian-arabic-translator")
+model = AutoModelForSeq2SeqLM.from_pretrained("Omar-youssef/english-egyptian-arabic-translator")
+
+def translate(text):
+    inputs = tokenizer.encode(text, return_tensors="pt")
+    outputs = model.generate(inputs, num_beams=4, early_stopping=True)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+print(translate("We haven't seen each other in a long time. We should catch up soon."))
+# Output: إحنا مش بنشوف بعض من فترة طويلة. المفروض نلحق ببعضنا قريب.
+
+print(translate("It's getting cold outside, so take a jacket with you."))
+# Output: الجو طلع برد بره، فاخد جاكيت معاك.
 ```
 
 ## Training Details