Update app.py

app.py

@@ -51,5 +51,10 @@ def predict_segmentation(img):
 input_image = gr.inputs.Image()
 output_image1 = gr.outputs.Image(type='pil')
 output_image2 = gr.outputs.Image(type='pil')
-app = gr.Interface(fn=predict_segmentation, inputs=input_image, outputs=[output_image1, output_image2], title='Microstructure Segmentation', description='Segment the input image into grain and background.')
+# Define the example image
+example_image = './albumentations/original/micrograph20.jpeg'
+example_input = [example_image]
+example_output = predict_segmentation(example_image)
+examples = [(example_input, example_output)]
+app = gr.Interface(fn=predict_segmentation, inputs=input_image, outputs=[output_image1, output_image2], title='Microstructure Segmentation', description='Segment the input image into grain and background.', examples=examples)
 app.launch()