Update app.py

app.py

@@ -1,8 +1,7 @@
 import json
 import gradio as gr
-from datasets import load_dataset
-
-ds = load_dataset("tti-bias/identities", split="train")
+import os
+from PIL import Image
 
 clusters_12 = json.load(open("clusters/id_all_blip_clusters_12.json"))
 clusters_24 = json.load(open("clusters/id_all_blip_clusters_12.json"))
@@ -16,11 +15,12 @@ clusters_by_size = {
 
 def show_cluster(cl_id, num_clusters):
     cl_dct = clusters_by_size[num_clusters][cl_id]
-    img_paths = []
-    for i in range(5):
+    images = []
+    for i in range(6):
         img_path = "/".join([st.replace("/", "") for st in cl_dct['img_path_list'][i].split("//")][3:])
-        img_paths.append(img_path)
-    return len(cl_dct['img_path_list']),cl_dct["labels_gender"], cl_dct["labels_model"], cl_dct["labels_ethnicity"], img_paths
+        img_path = os.path.join("identities-images", img_path)
+        images.append(Image.open(img_path))
+    return len(cl_dct['img_path_list']),cl_dct["labels_gender"], cl_dct["labels_model"], cl_dct["labels_ethnicity"], images
 
-demo = gr.Interface(fn=show_cluster, inputs=[gr.Slider(0, 50), gr.Radio([12, 24, 48])], outputs=["text", "text", "text", "text", "text"])
+demo = gr.Interface(fn=show_cluster, inputs=[gr.Slider(0, 50), gr.Radio([12, 24, 48])], outputs=["text", "text", "text", "text", gr.Gallery(3)])
 demo.launch(debug=True)