Working now

.idea/.gitignore

@@ -0,0 +1,3 @@
+# Default ignored files
+/shelf/
+/workspace.xml

.idea/DiffusionFaceClustering.iml

@@ -0,0 +1,10 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<module type="PYTHON_MODULE" version="4">
+  <component name="NewModuleRootManager">
+    <content url="file://$MODULE_DIR$">
+      <excludeFolder url="file://$MODULE_DIR$/venv" />
+    </content>
+    <orderEntry type="inheritedJdk" />
+    <orderEntry type="sourceFolder" forTests="false" />
+  </component>
+</module>

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,47 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredPackages">
+        <value>
+          <list size="27">
+            <item index="0" class="java.lang.String" itemvalue="numpy" />
+            <item index="1" class="java.lang.String" itemvalue="nltk" />
+            <item index="2" class="java.lang.String" itemvalue="transformers" />
+            <item index="3" class="java.lang.String" itemvalue="streamlit" />
+            <item index="4" class="java.lang.String" itemvalue="huggingface_hub" />
+            <item index="5" class="java.lang.String" itemvalue="datasets" />
+            <item index="6" class="java.lang.String" itemvalue="tokenizers" />
+            <item index="7" class="java.lang.String" itemvalue="pyarrow" />
+            <item index="8" class="java.lang.String" itemvalue="scipy" />
+            <item index="9" class="java.lang.String" itemvalue="dataclasses" />
+            <item index="10" class="java.lang.String" itemvalue="iso639" />
+            <item index="11" class="java.lang.String" itemvalue="python_igraph" />
+            <item index="12" class="java.lang.String" itemvalue="scikit-learn" />
+            <item index="13" class="java.lang.String" itemvalue="langdetect" />
+            <item index="14" class="java.lang.String" itemvalue="PyYAML" />
+            <item index="15" class="java.lang.String" itemvalue="scikit_learn" />
+            <item index="16" class="java.lang.String" itemvalue="plotly" />
+            <item index="17" class="java.lang.String" itemvalue="torch" />
+            <item index="18" class="java.lang.String" itemvalue="streamlit-aggrid" />
+            <item index="19" class="java.lang.String" itemvalue="pandas" />
+            <item index="20" class="java.lang.String" itemvalue="tqdm" />
+            <item index="21" class="java.lang.String" itemvalue="iso_639" />
+            <item index="22" class="java.lang.String" itemvalue="powerlaw" />
+            <item index="23" class="java.lang.String" itemvalue="seaborn" />
+            <item index="24" class="java.lang.String" itemvalue="sentencepiece" />
+            <item index="25" class="java.lang.String" itemvalue="matplotlib" />
+            <item index="26" class="java.lang.String" itemvalue="numexpr" />
+          </list>
+        </value>
+      </option>
+    </inspection_tool>
+    <inspection_tool class="PyUnresolvedReferencesInspection" enabled="true" level="WARNING" enabled_by_default="true">
+      <option name="ignoredIdentifiers">
+        <list>
+          <option value="list.isnull" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,4 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (DiffusionFaceClustering)" project-jdk-type="Python SDK" />
+</project>

.idea/modules.xml

@@ -0,0 +1,8 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ProjectModuleManager">
+    <modules>
+      <module fileurl="file://$PROJECT_DIR$/.idea/DiffusionFaceClustering.iml" filepath="$PROJECT_DIR$/.idea/DiffusionFaceClustering.iml" />
+    </modules>
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

app.py

@@ -14,27 +14,38 @@ clusters_by_size = {
 }
 
 def show_cluster(cl_id, num_clusters):
+    if not cl_id:
+        cl_id = 0
+    if not num_clusters:
+        num_clusters = 12
     cl_dct = clusters_by_size[num_clusters][cl_id]
     images = []
     for i in range(6):
         img_path = "/".join([st.replace("/", "") for st in cl_dct['img_path_list'][i].split("//")][3:])
         images.append((Image.open(os.path.join("identities-images", img_path)), "_".join([img_path.split("/")[0], img_path.split("/")[-1]])))
     return (len(cl_dct['img_path_list']),
-           dict(cl_dct["labels_gender"]),
+            dict(cl_dct["labels_gender"]),
             dict(cl_dct["labels_model"]),
             dict(cl_dct["labels_ethnicity"]),
             images)
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Cluster Explorer")
-    gr.HTML("""<span style="color:red">⚠️ <b>DISCLAIMER: the images displayed by this tool were generated by text-to-image models and may depict offensive stereotypes or contain explicit content.</b></span>""")
+    gr.Markdown("# Diffusion Faces Cluster Explorer")
+    gr.HTML("""<span><b>This Space lets you explore the data generated from <a href="https://huggingface.co/spaces/society-ethics/DiffusionBiasExplorer" target=_blank>DiffusionBiasExplorer</a></b></span>.""")
+    gr.HTML("""<span style="color:red">⚠️ DISCLAIMER: the images displayed by this tool were generated by text-to-image models and may depict offensive stereotypes or contain explicit content.</span>""")
     with gr.Row():
-        num_clusters = gr.Radio([12, 24, 48], label="number of clusters")
-        cluster_id = gr.Number(precision=0, label="cluster id")
-        button = gr.Button(value="Go")
+        num_clusters = gr.Radio([12, 24, 48], label="How many clusters do you want to make from the data?")
+    with gr.Row():
+        with gr.Column():
+            button = gr.Button(value="Go!")
+        with gr.Column():
+            pass
     with gr.Column():
         gallery = gr.Gallery(label="Most representative images in cluster").style(grid=6)
-        a = gr.Text(label="Number of items in cluster")
+        with gr.Row():
+            with gr.Column():
+                cluster_id = gr.Slider(0, 20, step=1, label="Cluster number")
+                a = gr.Text(label="Number of images")
         with gr.Row():
             c = gr.Text(label="Model makeup of cluster")
             b = gr.Text(label="Gender label makeup of cluster")

requirements.txt

@@ -1 +1,2 @@
-datasets[vision]
+datasets[vision]
+gradio