Update README.md

README.md

@@ -12,21 +12,25 @@ tags:
 - Simplification
 - text-to-text
 ---
-# Persian Simplification Model (ParsT5 Base)
+# Persian Simplification Model (parsT5 Base)
 
 ---
 
 ## Overview
 
-This model is a fine-tuned version of ParsT5 (base) designed explicitly for the Persian Simplification Task. The training data consists of Persian legal texts. The model is trained using supervised fine-tuning and employs the **Unlimiformer Algorithm** to handle large inputs effectively.
+This model is a fine-tuned ParsT5 (base) version designed explicitly for the Persian Simplification Task. The training data consists of Persian legal texts. The model is trained using supervised fine-tuning and employs the **Unlimiformer Algorithm** to handle large inputs effectively.
 
-- **Architecture**: ParsT5-base
+- **Architecture**: Ahmad/parsT5-base
 - **Language**: Persian
 - **Task**: Text Simplification
 - **Training Setup**:
   - **Algorithm for reducing computation**: Unlimiformer
   - **Epochs**: 12
   - **Hardware**: NVIDIA GPU 4070
+  - **Trainable Blocks**: Last Encoder-Decoder
+  - **Optimizer** : AdamW + lr_scheduler
+  - **Input max Tokens**: 4096
+  - **Output max Tokens**: 512
 
 ---
 
@@ -45,7 +49,7 @@ The following table summarizes the readability scores for the original texts and
 
 ## Evaluation Results
 
-The fine-tuned model was evaluated using **Rouge** and **BERTScore** metrics. For comparison, the performance of two other Persian LLMs based on LLaMA is also presented:
+The fine-tuned model was evaluated using **Rouge** and **BERTScore (mBERT)** metrics. For comparison, the performance of two other Persian LLMs based on LLaMA is also presented:
  
 
 | Prediction Model                              | Rouge1  | Rouge2  | RougeL  | Precision | Recall  | F1      |