RunningYou commited on
Commit
96a0d26
1 Parent(s): 633ceb7
Files changed (1) hide show
  1. app.py +10 -6
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
- # 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,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).launch(enable_queue=True)
 
 
 
 
 
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)