Update app.py

app.py

@@ -4,13 +4,13 @@ from datasets import load_dataset
 import random
 
 auth_token = os.environ.get("auth_token")
-iiw_400 = load_dataset('google/imageinwords', token=auth_token, name="IIW-400")
+iiw_400 = load_dataset('google/imageinwords', token=auth_token, trust_remote_code=True, name="IIW-400")
 
 def display_iiw_data(index):
     data = iiw_400['test'][index]
     image_html = f'<img src="{data["image/url"]}" style="width:100%; max-width:800px; height:auto;">'
-    iiw_text = f"<h2>IIW Human Descriptions:</h2><p style='font-size: 16px'>{data['IIW']}</p>"
-    iiw_p5b_text = f"<h2>IIW PaLI 5B Predictions:</h2><p style='font-size: 16px'>{data['IIW-P5B']}</p>"
+    iiw_text = f"<h2>IIW Human-Authored Descriptions:</h2><p style='font-size: 16px'>{data['IIW']}</p>"
+    iiw_p5b_text = f"<h2>IIW PaLI-5B Generated Descriptions:</h2><p style='font-size: 16px'>{data['IIW-P5B']}</p>"
     ratings = "<h2>Ratings:</h2>"
     if data['iiw-human-sxs-iiw-p5b'] is not None:
         for key, value in data['iiw-human-sxs-iiw-p5b'].items():