Update app.py

app.py

@@ -193,9 +193,11 @@ def image_click(images, evt: gr.SelectData,
 
 
 with gr.Blocks() as demo:
-    with gr.Row():
-        choose_model = gr.Radio(choices=["0", "1", "2", "3"],
-        value="0", label="Choose embedding layer", elem_id="layer_radio")
+    title = gr.HTML(
+            """<h1><img src="https://i.imgur.com/52VJ8vS.png" alt="SD"> Deepsparse Image Embedding </h1>""",
+            elem_id="title",
+    )
+
     with gr.Row():
         with gr.Column():
             inputs_0 = gr.Image(label = "Input Image for embed")
@@ -205,6 +207,15 @@ with gr.Blocks() as demo:
             button_1 = gr.Button("Image File button")
     with gr.Row():
         with gr.Column():
+            with gr.Row():
+                title = gr.Markdown(
+                    value="### Click on a Image in the gallery to select it",
+                    visible=True,
+                    elem_id="selected_model",
+                )
+                choose_model = gr.Radio(choices=["0", "1", "2", "3"],
+            value="0", label="Choose embedding layer", elem_id="layer_radio")
+
             g_outputs = gr.Gallery(label='Output gallery', elem_id="gallery",
             columns=[5],object_fit="contain", height="auto")
             outputs = gr.Text(label = "Output Embeddings")
@@ -214,6 +225,23 @@ with gr.Blocks() as demo:
             sg_outputs = gr.Gallery(label='Sort gallery', elem_id="gallery",
             columns=[5],object_fit="contain", height="auto")
 
+    with gr.Row():
+        gr.Examples(
+                [
+                    "Anything_V5.png",
+                    "waifu_girl0.png",
+                ],
+                inputs = inputs_0,
+                label = "Image Examples"
+        )
+        gr.Examples(
+                [
+                    "rose_love_imgs.zip",
+                    "beautiful_room_imgs.zip"
+                ],
+                inputs = inputs_1
+                label = "Image Zip file Examples"
+        )
 
     button_0.click(fn = emb_img_func, inputs = [inputs_0, choose_model], outputs = outputs)
     button_1.click(fn = unzip_ims_func, inputs = [inputs_1, choose_model],
@@ -223,6 +251,4 @@ with gr.Blocks() as demo:
         inputs = [g_outputs, choose_model],
         outputs = [sdg_outputs, sg_outputs],)
 
-
-
 demo.launch("0.0.0.0")