Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
22 |
-
|
23 |
-
|
24 |
submit_btn = gr.Button("Submit")
|
25 |
-
|
|
|
26 |
with gr.Tab("ControlNet with Semantic Segmentation"):
|
27 |
-
|
28 |
-
|
29 |
-
|
30 |
submit_btn = gr.Button("Submit")
|
31 |
-
|
|
|
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()
|