Update data_reviewer.py

data_reviewer.py

@@ -42,9 +42,6 @@ def load_and_display_sample(split, idx):
 
         sample = dataset[idx]
 
-        # Process image
-        image = base64_to_image(sample["image"])
-
         # Get responses
         chosen_response, rejected_response = get_responses(sample["response"], sample["human_ranking"])
 
@@ -58,7 +55,7 @@ def load_and_display_sample(split, idx):
         )
 
         return (
-            image,  # image
+            sample["image"],  # image
             sample["id"],  # sample_id
             chosen_response,  # chosen_response
             rejected_response,  # rejected_response