Update app.py

app.py

@@ -35,7 +35,9 @@ img#style-image {
   max-height: 600px;
 }
 '''
-examples = [f"example{i}.jpg" for i in range(1,7)]
+def set_example_image(example: list) -> dict:
+    return gr.Image.update(value=example[0])
+    
 demo = gr.Blocks(css=css)
 with demo:
   gr.Markdown('''<h1 id="title">Image Captioning 🖼️</h1>''')
@@ -44,6 +46,6 @@ with demo:
         output = gr.outputs.Textbox(type="auto",label="Captions")
         examples = examples
   btn = gr.Button("Generate Caption")
-  btn.click(fn=predict, inputs=input, outputs=output)
+  btn.click(fn=predict, inputs=input, outputs=output, examples = examples)
 
 demo.launch()