Improve model card: Add pipeline tag and setup instructions

README.md

@@ -22,96 +22,97 @@ model-index:
       name: LOL-v1
       type: LOL-v1
     metrics:
-    - name: PSNR
-      type: PSNR
+    - type: PSNR
       value: 27.97
-    - name: SSIM
-      type: SSIM
+      name: PSNR
+    - type: SSIM
       value: 0.897
+      name: SSIM
   - task:
       type: low-light-image-enhancement
     dataset:
       name: LOL-v2-Real
       type: LOL-v2-Real
     metrics:
-    - name: PSNR
-      type: PSNR
+    - type: PSNR
       value: 29.33
-    - name: SSIM
-      type: SSIM
+      name: PSNR
+    - type: SSIM
       value: 0.915
+      name: SSIM
   - task:
       type: low-light-image-enhancement
     dataset:
       name: LOL-v2-Synthetic
       type: LOL-v2-Synthetic
     metrics:
-    - name: PSNR
-      type: PSNR
+    - type: PSNR
       value: 30.15
-    - name: SSIM
-      type: SSIM
+      name: PSNR
+    - type: SSIM
       value: 0.951
+      name: SSIM
   - task:
       type: low-light-image-enhancement
     dataset:
       name: SDSD-indoor
       type: SDSD-indoor
     metrics:
-    - name: PSNR
-      type: PSNR
+    - type: PSNR
       value: 31.37
-    - name: SSIM
-      type: SSIM
+      name: PSNR
+    - type: SSIM
       value: 0.917
+      name: SSIM
   - task:
       type: low-light-image-enhancement
     dataset:
       name: SDSD-outdoor
       type: SDSD-outdoor
     metrics:
-    - name: PSNR
-      type: PSNR
+    - type: PSNR
       value: 31.73
-    - name: SSIM
-      type: SSIM
+      name: PSNR
+    - type: SSIM
       value: 0.904
+      name: SSIM
   - task:
       type: low-light-image-enhancement
     dataset:
       name: MEF
       type: MEF
     metrics:
-    - name: NIQE
-      type: NIQE
+    - type: NIQE
       value: 3.44
+      name: NIQE
   - task:
       type: low-light-image-enhancement
     dataset:
       name: LIME
       type: LIME
     metrics:
-    - name: NIQE
-      type: NIQE
+    - type: NIQE
       value: 3.82
+      name: NIQE
   - task:
       type: low-light-image-enhancement
     dataset:
       name: DICM
       type: DICM
     metrics:
-    - name: NIQE
-      type: NIQE
+    - type: NIQE
       value: 3.64
+      name: NIQE
   - task:
       type: low-light-image-enhancement
     dataset:
       name: NPE
       type: NPE
     metrics:
-    - name: NIQE
-      type: NIQE
+    - type: NIQE
       value: 3.55
+      name: NIQE
+pipeline_tag: image-to-image
 ---
 
 # ✨ ModalFormer: Multimodal Transformer for Low-Light Image Enhancement
@@ -120,7 +121,7 @@ model-index:
   
 **[Alexandru Brateanu](https://scholar.google.com/citations?user=ru0meGgAAAAJ&hl=en), [Raul Balmez](https://scholar.google.com/citations?user=vPC7raQAAAAJ&hl=en), [Ciprian Orhei](https://scholar.google.com/citations?user=DZHdq3wAAAAJ&hl=en), [Codruta Ancuti](https://scholar.google.com/citations?user=5PA43eEAAAAJ&hl=en), [Cosmin Ancuti](https://scholar.google.com/citations?user=zVTgt8IAAAAJ&hl=en)**
 
-[![arXiv](https://img.shields.io/badge/arxiv-paper-179bd3)](https://arxiv.org/abs/2401.15204)
+[![arXiv](https://img.shields.io/badge/arxiv-paper-179bd3)](https://arxiv.org/abs/2507.20388)
 </div>
 
 ### Abstract
@@ -131,7 +132,25 @@ model-index:
 - `28.07.2025` 📦 Pre-trained models and test data published! ArXiv paper version and HuggingFace demo coming soon, stay tuned!
 
 ## ⚙️ Setup and Testing
-Please check out the [**GitHub repository**](https://github.com/albrateanu/ModalFormer) for implementation details.
+For ease, utilize a Linux machine with CUDA-ready devices (GPUs).
+
+To setup the environment, first run the provided setup script:
+
+```bash
+./environment_setup.sh
+# or 
+bash environment_setup.sh
+```
+
+Note: in case of difficulties, ensure ```environment_setup.sh``` is executable by running:
+
+```bash
+chmod +x environment_setup.sh
+```
+
+Give the setup a couple of minutes to run.
+
+Please check out the [**GitHub repository**](https://github.com/albrateanu/ModalFormer) for more implementation details.
 
 ## 📚 Citation