Update app.py

app.py

@@ -22,5 +22,19 @@ def show_cluster(cl_id, num_clusters):
         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", gr.Gallery()])
+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>""")
+    with gr.Row():
+        num_clusters = gr.Radio([12, 24, 48])
+        cluster_id = gr.Slider(0, len(num_clusters))
+        button = gr.Button(value="Go")
+    with gr.Column():
+        a = gr.Text()
+        b = gr.Text()
+        c = gr.Text()
+        d = gr.Text()
+    gallery = gr.Gallery().style(grid=4)
+    button.click(fn=show_cluster, inputs=[num_clusters, cluster_id], outputs=[gr.Text(),gr.Text(),gr.Text(),gr.Text(), gallery])
+# demo = gr.Interface(fn=show_cluster, inputs=[gr.Slider(0, 50), gr.Radio([12, 24, 48])], outputs=["text", "text", "text", "text", gr.Gallery()])
 demo.launch(debug=True)