merve HF Staff commited on
Commit
4486ac5
·
1 Parent(s): cdcd05b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -8
app.py CHANGED
@@ -18,16 +18,18 @@ with gr.Blocks() as demo:
18
 
19
  input_list = [prompt_input, negative_prompt_input, image_input]
20
  with gr.Tab("ControlNet on Canny Filter "):
21
- prompt_input = gr.Textbox(label="Prompt")
22
- negative_prompt_input = gr.Textbox(label="Negative Prompt")
23
- image_input = gr.Image(label="Image")
24
  submit_btn = gr.Button("Submit")
25
- submit_btn.click(fn=write_to_dataset, inputs=input_list, outputs="image")
 
26
  with gr.Tab("ControlNet with Semantic Segmentation"):
27
- prompt_input = gr.Textbox(label="Prompt")
28
- negative_prompt_input = gr.Textbox(label="Negative Prompt")
29
- image_input = gr.Image(label="Image")
30
  submit_btn = gr.Button("Submit")
31
- submit_btn.click(fn=write_to_dataset, inputs=input_list, outputs="image")
 
32
 
33
  demo.launch()
 
18
 
19
  input_list = [prompt_input, negative_prompt_input, image_input]
20
  with gr.Tab("ControlNet on Canny Filter "):
21
+ prompt_input_canny = gr.Textbox(label="Prompt")
22
+ negative_prompt_canny = gr.Textbox(label="Negative Prompt")
23
+ canny_input = gr.Image(label="Image")
24
  submit_btn = gr.Button("Submit")
25
+ canny_inputs = [prompt_input_canny, negative_prompt_canny, canny_input]
26
+ submit_btn.click(fn=write_to_dataset, inputs=canny_inputs, outputs="image")
27
  with gr.Tab("ControlNet with Semantic Segmentation"):
28
+ prompt_input_seg = gr.Textbox(label="Prompt")
29
+ negative_prompt_seg = gr.Textbox(label="Negative Prompt")
30
+ seg_input = gr.Image(label="Image")
31
  submit_btn = gr.Button("Submit")
32
+ seg_inputs = [prompt_input_seg, negative_prompt_seg, seg_input]
33
+ submit_btn.click(fn=write_to_dataset, inputs=seg_inputs, outputs="image")
34
 
35
  demo.launch()