Update README.md

README.md

@@ -5,24 +5,24 @@ language:
 pipeline_tag: text-generation
 tags:
 - Arabic_Dialectal_Lemmatization
----
-
+---
+
 # Gulf-S2S-lemmatizer
-
-# Model Description
-
-The model is developed for Arabic dialect lemmatization, focusing on Gulf (GLF) Arabic. It follows a sequence-to-sequence formulation of lemmatization, where the model generates the lemma of a given word knowing 2 words before and 2 words after the current word rather than treating lemmas as fixed classification labels.
-
-The model is evaluated using lemma accuracy as the main metric, with an additional normalized lemma accuracy metric that accounts for orthographic and diacritic variation. The full methodology, training setup, hyperparameters, and evaluation results are described in our paper [“Lemmatizing Dialectal Arabic with Sequence-to-Sequence Models”](https://aclanthology.org/2025.arabicnlp-main.10/)
-
-
+
+# Model Description
+
+The model is developed for Arabic dialect lemmatization, focusing on Gulf (GLF) Arabic. It follows a sequence-to-sequence formulation of lemmatization, where the model generates the lemma of a given word knowing 2 words before and 2 words after the current word rather than treating lemmas as fixed classification labels.
+
+The model is evaluated using lemma accuracy as the main metric, with an additional normalized lemma accuracy metric that accounts for orthographic and diacritic variation. The full methodology, training setup, hyperparameters, and evaluation results are described in our paper [“Lemmatizing Dialectal Arabic with Sequence-to-Sequence Models”](https://aclanthology.org/2025.arabicnlp-main.10/)
+
+
 # Standalone Usage
 
 
-
+
 The model can also be used independently without the full lemmatization workflow on the GitHub repository (https://github.com/CAMeL-Lab/seq2seq-arabic-dialect-lemmatization). In this case, the input should contain the target word surrounded by the special token `<target>`, with up to two words before and two words after the target word.
-
-
+
+
 ```python
 import re
 import math
@@ -33,7 +33,7 @@ from transformers import T5Tokenizer, T5ForConditionalGeneration
 
 DIALECT_MODELS = {
     "glf": "CAMeL-Lab/GLF-S2S-lemmatizer",
-}
+}
 def load_model(s2s_dialect: str):
     model_name = DIALECT_MODELS[s2s_dialect]
     tokenizer = T5Tokenizer.from_pretrained(model_name, use_fast=True, legacy=False)
@@ -82,16 +82,16 @@ def predict(tokenizer, model, texts: list[str], device=None, batch_size: int = 1
 
 # Example:
 # df should contain an input_text column with the target word marked using <target>
-# Example input: "أنا عايز <target>أروح<target> البيت دلوقتي"
+# Example input: "أنا أبي <target>أروح<target> البيت الحين"
 
 df["word"] = df["input_text"].apply(extract_target_word)
 
 tokenizer, model = load_model("egy")
 df["predicted_lex"] = predict(tokenizer, model, df["input_text"].tolist())
-```
-
-## 📖 Citation
-
+```
+
+## 📖 Citation
+
 If you use this model in your research, please cite the following paper:
 
 ```bibtex
@@ -103,5 +103,5 @@ If you use this model in your research, please cite the following paper:
     address = {Suzhou, China},
     url = {https://aclanthology.org/2025.arabicnlp-main.10/}
 }
-```
-
+```
+