Update app.py

app.py

@@ -16,7 +16,16 @@ clusters_by_size = {
 
 def show_cluster(cl_id, num_clusters):
     cl_dct = clusters_by_size[num_clusters][cl_id]
-    return len(cl_dct['img_path_list']),cl_dct["labels_gender"], cl_dct["labels_model"], cl_dct["labels_ethnicity"], ["/".join(img_path.replace("/", "")) for img_path in cl_dct["img_path_list"][:5]]
+    img_paths = []
+    for i in range(5):
+        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
+
+    for i in range(5):
+        img_path = "/".join([st.replace("/", "") for st in cl_dct['img_path_list'][i].split("//")][3:])
+        display(Image.open(pjoin(_DATA_PATH, img_path)).convert('RGB'))
+
 
 demo = gr.Interface(fn=show_cluster, inputs=[gr.Slider(0, 50), gr.Radio([12, 24, 48])], outputs=["text", "text", "text", "text", "text"])
 demo.launch(debug=True)