Update README.md

README.md

@@ -1,7 +1,12 @@
+---
+license: apache-2.0
+language:
+- ar
+---
 
 # Surya OCR Arabic
 
-This repository contains the `surya-ocr-arabic` model, which is based on a modified SegFormer architecture. The model was fine-tuned for document layout detection tasks.
+This repository contains the `surya-ocr-arabic-segment` model, which is based on a modified SegFormer architecture. The model was fine-tuned for document segmentation tasks.
 
 
 ## Setup Instructions
@@ -11,7 +16,7 @@ This repository contains the `surya-ocr-arabic` model, which is based on a modif
 To use the `SegformerForRegressionMask` class, you need to clone the Surya OCR GitHub repository:
 
 ```bash
-git clone https://github.com/vikp/surya_.git
+git clone https://github.com/vikp/surya.git
 cd surya
 ```
 
@@ -31,12 +36,14 @@ pip install -r requirements.txt
 
 ### Import and Use the Model
 
-You can load and use the `surya-ocr-arabic` model as follows:
+You can load and use the `surya-ocr-arabic-segment` model as follows:
 
 ```python
+
+#we are importing `SegformerForRegressionMask` from the folder of surya OCR repo.
 from surya.surya.model.detection.segformer import SegformerForRegressionMask
 import torch
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = SegformerForRegressionMask.from_pretrained("ketanmore/surya-ocr-arabic-segment", torch_dtype=torch.float32).to(device)
-```
+```