Spaces:
Runtime error
Runtime error
Commit
•
06cac1e
1
Parent(s):
f6c4ee3
mask
Browse files
app.py
CHANGED
@@ -106,10 +106,12 @@ image_input = gr.Image(source='upload', tool='sketch', type='pil')
|
|
106 |
prompt = gr.Textbox(label='prompt')
|
107 |
number = gr.Slider(1, 4, value=2, label='num_samples')
|
108 |
|
109 |
-
|
|
|
|
|
110 |
greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
|
111 |
greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
|
112 |
-
demo = gr.Parallel(greeter_1, greeter_2)
|
113 |
|
114 |
if __name__ == "__main__":
|
115 |
demo.launch(enable_queue=True)
|
|
|
106 |
prompt = gr.Textbox(label='prompt')
|
107 |
number = gr.Slider(1, 4, value=2, label='num_samples')
|
108 |
|
109 |
+
init = {'image': PIL.Image.open(os.path.join(os.path.dirname(__file__), 'example1.png')),
|
110 |
+
'mask': PIL.Image.open(os.path.join(os.path.dirname(__file__), 'example1_mask.png'))}
|
111 |
+
examples = [[init, 'a bench in a field', 2]]
|
112 |
greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
|
113 |
greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
|
114 |
+
demo = gr.Parallel(greeter_1, greeter_2, examples=examples)
|
115 |
|
116 |
if __name__ == "__main__":
|
117 |
demo.launch(enable_queue=True)
|