mask

app.py

@@ -108,10 +108,10 @@ number = gr.Slider(1, 4, value=2, label='num_samples')
 #
 # init = {'image': PIL.Image.open(os.path.join(os.path.dirname(__file__), 'example1.png')),
 #         'mask': PIL.Image.open(os.path.join(os.path.dirname(__file__), 'example1_mask.png'))}
-# examples = [[init, 'a bench in a field', 2]]
+examples = [[os.path.join(os.path.dirname(__file__), 'example1.png'), 'a bench in a field', 2]]
 greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
 greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
-demo = gr.Parallel(greeter_1, greeter_2)
+demo = gr.Parallel(greeter_1, greeter_2, examples=examples)
 
 if __name__ == "__main__":
     demo.launch(enable_queue=True)

requirements.txt

@@ -4,4 +4,5 @@ mediapipe
 scipy
 ftfy
 torch
-accelerate
+accelerate
+gradio