make example bigger / clearer

app.py

@@ -7,6 +7,7 @@ import numpy as np
 import pandas as pd
 import math
 from transformers import CLIPTextModel, CLIPTokenizer
+import os
 
 # model_id = "stabilityai/stable-diffusion-2-1-base"
 # text_model_id = "CompVis/stable-diffusion-v-1-4-original"
@@ -216,42 +217,39 @@ model = AutoModel.from_pretrained(clip_model_id)
 model = model.to(device)
 
 examples = [
-    ["frog.png", 3, 1, 25, 0],
-    ["img0.jpg", 3, 1, 25, 0],
-    ["img1.jpg", 3, 1, 25, 0],
-    ["img2.jpg", 3, 1, 25, 0],
-    ["img3.jpg", 3, 1, 25, 0],
+    ["SohoJoeEth.jpeg", "Ray-Liotta-Goodfellas.jpg", "SohoJoeEth + Ray.jpeg"],
+    # ["SohoJoeEth.jpeg", "Donkey.jpg", "SohoJoeEth + Donkey.jpeg"],
+    # ["SohoJoeEth.jpeg", "Snoop Dogg.jpg", "SohoJoeEth + Snoop Dogg.jpeg"],
 ]
+tile_size = 100
 
 
-with gr.Blocks() as demo:
-    with gr.Row():
-        gr.Markdown(
-"""# Soho-Clip
-""")
 
+with gr.Blocks() as demo:
     with gr.Row():
         with gr.Column(scale=5):
             gr.Markdown(
 """
+# Soho-Clip
+
 A tool for exploring CLIP embedding spaces.
 
-Try uploading a few images/add text prompts and click generate images.
+Try uploading a few images and/or add some text prompts and click generate images.
 """)
-        with gr.Column(scale=3):
+        with gr.Column(scale=2, min_width=(tile_size+20)*3):
             with gr.Row():
-                with gr.Column(scale=1, min_width=66):
-                    gr.Image(value = "SohoJoeEth.jpeg", shape=(66,66), show_label=False, interactive=False).style(height=66, width=66)
-                with gr.Column(scale=1, min_width=15):
-                    gr.Markdown("#   ")
-                    gr.Markdown("#   +")
-                with gr.Column(scale=1, min_width=66):
-                    gr.Image(value = "Ray-Liotta-Goodfellas.jpg", shape=(66,66), show_label=False, interactive=False).style(height=66, width=66)
-                with gr.Column(scale=1, min_width=15):
-                    gr.Markdown("#   ")
-                    gr.Markdown("#   =")
-                with gr.Column(scale=1, min_width=66):
-                    gr.Image(value = "SohoJoeEth + Ray.jpeg", shape=(66,66), show_label=False, interactive=False).style(height=66, width=66)
+                with gr.Column(scale=1, min_width=tile_size):
+                    gr.Markdown("## Input 1")
+                with gr.Column(scale=1, min_width=tile_size):
+                    gr.Markdown("## Input 2")
+                with gr.Column(scale=1, min_width=tile_size):
+                    gr.Markdown("## Generates:")
+            for example in examples:
+                with gr.Row():
+                    for image in example:
+                        with gr.Column(scale=1, min_width=tile_size):
+                            image_path = os.path.join("Images", image)
+                            gr.Image(value = image_path, shape=(tile_size,tile_size), show_label=False, interactive=False).style(height=tile_size, width=tile_size)
 
     with gr.Row():
         for i in range(max_tabs):