RunningYou commited on
Commit
03c2ca0
1 Parent(s): a35ff12
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -102,8 +102,8 @@ def predict2(dict, prompt, num_samples):
102
  return image
103
 
104
 
105
- image_input = gr.Image(source='upload', type='pil') # tool='sketch',
106
- prompt = gr.Textbox(label='prompt')
107
  number = gr.Slider(1, 4, value=2, label='num_samples')
108
 
109
  examples = [
@@ -114,7 +114,7 @@ examples = [
114
 
115
  greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
116
  greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
117
- demo = gr.Parallel(greeter_1, greeter_2, examples=examples)
118
 
119
  if __name__ == "__main__":
120
  demo.launch(enable_queue=True)
 
102
  return image
103
 
104
 
105
+ image_input = gr.Image(source='upload', tool='sketch', type='pil', value=os.path.join(os.path.dirname(__file__), 'example1.png'))
106
+ prompt = gr.Textbox(label='prompt', value='a bench in a field')
107
  number = gr.Slider(1, 4, value=2, label='num_samples')
108
 
109
  examples = [
 
114
 
115
  greeter_1 = gr.Interface(predict1, inputs=[image_input, prompt, number], outputs=gr.Image(label='auto'))
116
  greeter_2 = gr.Interface(predict2, inputs=[image_input, prompt, number], outputs=gr.Image(label='paint'))
117
+ demo = gr.Parallel(greeter_1, greeter_2)
118
 
119
  if __name__ == "__main__":
120
  demo.launch(enable_queue=True)