Update app.py

app.py

@@ -35,13 +35,17 @@ def search(image_dict):
     # Get similar image
     scores, retrieved_examples = get_neighbors(query_image)
 
+    final_md = ""
+    
     # Create result diction for gr.Gallery
     result = []
     for i in range(len(retrieved_examples["image"])):
         name = retrieved_examples["name"][i]
         result.append((retrieved_examples["image_link"][i], name))
+
+        final_md += """![](""" + retrieved_examples["image_link"][i] + """)\n"""
     
-    return result
+    return result, final_md
 
 iface = gr.Interface(fn=search,
                      description="""
@@ -51,6 +55,6 @@ iface = gr.Interface(fn=search,
                      <br>
                      """,
                      inputs=gr.ImageEditor(label='Sketchpad' ,type='filepath', value={'background': './template.JPG', 'layers': None, 'composite': None}, sources=['upload'], transforms=[]), 
-                     outputs=gr.Gallery(label='Similar', object_fit='contain', height=1200),
+                     outputs=[gr.Gallery(label='Similar', object_fit='contain', height=1200), gr.Markdown()],
                      theme = gr.themes.Base(primary_hue="teal",secondary_hue="teal",neutral_hue="slate"),)
 iface.launch()