Spaces:
Runtime error
Runtime error
Commit
•
96a0d26
1
Parent(s):
633ceb7
timeout
Browse files
app.py
CHANGED
@@ -84,11 +84,11 @@ title = "Person Matting & Stable Diffusion In-Painting"
|
|
84 |
description = "Inpainting Stable Diffusion <br/><b>mediapipe + Stable Diffusion<b/><br/>"
|
85 |
|
86 |
|
87 |
-
|
88 |
-
|
89 |
-
#
|
90 |
-
#
|
91 |
-
|
92 |
|
93 |
|
94 |
def predict1(dict, prompt, num_samples):
|
@@ -112,4 +112,8 @@ number = gr.Slider(1, 4, value= 2, label='num_samples')
|
|
112 |
|
113 |
greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
|
114 |
greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
|
115 |
-
demo = gr.Parallel(greeter_1, greeter_2
|
|
|
|
|
|
|
|
|
|
84 |
description = "Inpainting Stable Diffusion <br/><b>mediapipe + Stable Diffusion<b/><br/>"
|
85 |
|
86 |
|
87 |
+
examples = [
|
88 |
+
[os.path.dirname(__file__) + 'example1.png', 'a bench in a field', 2],
|
89 |
+
# [os.path.join(os.path.dirname(__file__), "example2.png"), 'a building with many steps', 2],
|
90 |
+
# [os.path.join(os.path.dirname(__file__), "example3.png"), 'a big ship parked on the shore', 2]
|
91 |
+
]
|
92 |
|
93 |
|
94 |
def predict1(dict, prompt, num_samples):
|
|
|
112 |
|
113 |
greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
|
114 |
greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
|
115 |
+
demo = gr.Parallel(greeter_1, greeter_2, examples=examples)
|
116 |
+
|
117 |
+
|
118 |
+
if __name__ == "__main__":
|
119 |
+
demo.launch(enable_queue=True)
|