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():