Update app.py

app.py

@@ -20,7 +20,11 @@ def show_cluster(cl_id, num_clusters):
         img_path = "/".join([st.replace("/", "") for st in cl_dct['img_path_list'][i].split("//")][3:])
         img_path = os.path.join("identities-images", img_path)
         images.append(Image.open(img_path))
-    return len(cl_dct['img_path_list']), dict(cl_dct["labels_gender"]), dict(cl_dct["labels_model"]), dict(cl_dct["labels_ethnicity"]), images
+    return (len(cl_dct['img_path_list']),
+            json.dumps(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")
@@ -31,9 +35,10 @@ with gr.Blocks() as demo:
         button = gr.Button(value="Go")
     with gr.Column():
         a = gr.Text(label="Number of items in cluster")
-        c = gr.Text(label="Model makeup of cluster")
-        b = gr.Text(label="Gender label makeup of cluster")
-        d = gr.Text(label="Ethnicity label makeup of cluster")
+        with gr.Row():
+            c = gr.Text(label="Model makeup of cluster")
+            b = gr.Markdown(label="Gender label makeup of cluster")
+            d = gr.Text(label="Ethnicity label makeup of cluster")
     gallery = gr.Gallery(label="Most representative images in cluster").style(grid=6)
     button.click(fn=show_cluster, inputs=[cluster_id, num_clusters], outputs=[a,b,c,d, gallery])
 # demo = gr.Interface(fn=show_cluster, inputs=[gr.Slider(0, 50), gr.Radio([12, 24, 48])], outputs=["text", "text", "text", "text", gr.Gallery()])