Update app.py

app.py

@@ -118,8 +118,11 @@ with gr.Blocks(css=css) as block:
         
         submit_btn = gr.Button("Train")
         
-        image_output = gr.Gallery(label="Edited image")
-        #text_output = gr.Image(label="trained status")
+        image_output = gr.Image(label="Edited image")
+
+        examples=[['a sitting dog','imagic-dog.png'], ['a photo of a bird spreading wings','imagic-bird.png']]
+        ex = gr.Examples(examples=examples, fn=infer, inputs=[prompt_input,image_init], outputs=[image_output], cache_examples=False, run_on_click=True)
+        ex.dataset.headers = [""]
         
         gr.HTML(article)