resizing images

app.py

@@ -70,18 +70,10 @@ def show_cluster(cl_id, num_clusters):
         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]])
-                .replace("Photo_portrait_of_an_", "")
-                .replace("Photo_portrait_of_a_", "")
-                .replace("SD_v2_random_seeds_identity_", "(SD v.2) ")
-                .replace("dataset-identities-dalle2_", "(Dall-E 2) ")
-                .replace("SD_v1.4_random_seeds_identity_", "(SD v.1.4) ")
-                .replace("_", " "),
-            )
-        )
+        im = Image.open(os.path.join("identities-images", img_path))
+            #.resize((256, 256))
+        caption = "_".join([img_path.split("/")[0], img_path.split("/")[-1]]).replace("Photo_portrait_of_an_", "").replace("Photo_portrait_of_a_", "").replace("SD_v2_random_seeds_identity_", "(SD v.2) ").replace("dataset-identities-dalle2_", "(Dall-E 2) ").replace("SD_v1.4_random_seeds_identity_", "(SD v.1.4) ").replace("_", " ")             
+        images.append((im,caption))
     model_fig = go.Figure()
     model_fig.add_trace(
         go.Pie(
@@ -128,13 +120,8 @@ def show_cluster(cl_id, num_clusters):
 with gr.Blocks(title=TITLE) as demo:
     gr.Markdown(f"# {TITLE}")
     gr.Markdown(
-        "## Explore the data generated from [DiffusionBiasExplorer](https://huggingface.co/spaces/society-ethics/DiffusionBiasExplorer)!"
-    )
-    gr.Markdown(
-        "### This demo showcases patterns in the images generated from different prompts input to Stable Diffusion and Dalle-2 systems."
-    )
-    gr.Markdown(
-        "### Below, see results on how the images from different prompts cluster together."
+        "Explore the data generated from [DiffusionBiasExplorer](https://huggingface.co/spaces/society-ethics/DiffusionBiasExplorer)! This demo showcases patterns in the images generated from different prompts input to Stable Diffusion and Dalle-2 systems.")
+    gr.Markdown("See the results on how the images from different prompts cluster together below."
     )
     gr.HTML(
         """<span style="color:red" font-size:smaller>⚠️ DISCLAIMER: the images displayed by this tool were generated by text-to-image systems and may depict offensive stereotypes or contain explicit content.</span>"""
@@ -146,10 +133,8 @@ with gr.Blocks(title=TITLE) as demo:
     )
 
     with gr.Row():
-        with gr.Column(scale=4):
-            gallery = gr.Gallery(label="Most representative images in cluster").style(
-                grid=(3, 3)
-            )
+        with gr.Column():
+            gallery = gr.Gallery(label="Most representative images in cluster").style(grid=[2,3], height="auto")
         with gr.Column():
             cluster_id = gr.Dropdown(
                 choices=[i for i in range(num_clusters.value)],
@@ -158,7 +143,7 @@ with gr.Blocks(title=TITLE) as demo:
             )
             a = gr.Text(label="Number of images")
     with gr.Row():
-        with gr.Column(scale=1):
+        with gr.Column():
             c = gr.Plot(label="How many images from each system?")
             c_desc = gr.HTML(label="")
         with gr.Column(scale=1):