update

.gitattributes

@@ -33,3 +33,6 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+df_final.pkl filter=lfs diff=lfs merge=lfs -text
+batch_uPF4cMRPVfoUJk2d4Xgrypa8_output.jsonl filter=lfs diff=lfs merge=lfs -text
+gpt4o_mini_dataset.jsonl filter=lfs diff=lfs merge=lfs -text

app.py

@@ -1,6 +1,8 @@
 import gradio as gr
 import pandas as pd
 from datasets import load_dataset
+from PIL import Image, ImageOps
+
 
 df_final = pd.read_pickle("./df_final.pkl")
 dataset = load_dataset("XAI/vlmsareblind")
@@ -11,13 +13,23 @@ def show_row(row_index, selected_task):
     row = task_df.iloc[int(row_index)]
     custom_id = int(row["custom_id"])
     image = dataset["valid"][custom_id]["image"]
+
+    # Add white padding to the image
+    original_size = image.size
+    new_size = (original_size[0] * 2, original_size[1] * 2)
+    padding = (
+        (new_size[0] - original_size[0]) // 2,
+        (new_size[1] - original_size[1]) // 2,
+    )
+    image_with_padding = ImageOps.expand(image, border=padding, fill="white")
+
     prompt = dataset["valid"][custom_id]["prompt"]
     model_output = row["content_raw"]
     ground_truth = row["gt"]
     task = row["task"]
     is_correct = row["is_correct"]
 
-    return image, prompt, model_output, ground_truth, task, is_correct
+    return image_with_padding, prompt, model_output, ground_truth, task, is_correct
 
 
 def update_slider(selected_task):