Robert001 commited on
Commit
afa0370
1 Parent(s): c39a955

delete bbox

Browse files
Files changed (1) hide show
  1. app.py +30 -30
app.py CHANGED
@@ -1185,36 +1185,36 @@ with demo:
1185
  ddim_steps, guess_mode, strength, scale, seed, eta, condition_mode]
1186
  run_button.click(fn=process_seg, inputs=ips, outputs=[result_gallery])
1187
 
1188
- with gr.TabItem("Bbox"):
1189
- with gr.Row():
1190
- gr.Markdown("## UniControl Stable Diffusion with Object Bounding Boxes (MS-COCO)")
1191
- with gr.Row():
1192
- with gr.Column():
1193
- input_image = gr.Image(source='upload', type="numpy")
1194
- prompt = gr.Textbox(label="Prompt")
1195
- run_button = gr.Button(label="Run")
1196
- with gr.Accordion("Advanced options", open=False):
1197
- num_samples = gr.Slider(label="Images", minimum=1, maximum=12, value=1, step=1)
1198
- image_resolution = gr.Slider(label="Image Resolution", minimum=256, maximum=768, value=512,
1199
- step=64)
1200
- strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01)
1201
- condition_mode = gr.Checkbox(label='Condition Extraction: RGB -> Bbox', value=True)
1202
- guess_mode = gr.Checkbox(label='Guess Mode', value=False)
1203
- confidence = gr.Slider(label="Confidence of Detection", minimum=0.1, maximum=1.0, value=0.4,
1204
- step=0.1)
1205
- nms_thresh = gr.Slider(label="Nms Threshold", minimum=0.1, maximum=1.0, value=0.5, step=0.1)
1206
- ddim_steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=20, step=1)
1207
- scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=9.0, step=0.1)
1208
- seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, randomize=True)
1209
- eta = gr.Number(label="eta (DDIM)", value=0.0)
1210
- a_prompt = gr.Textbox(label="Added Prompt", value='best quality, extremely detailed, bright')
1211
- n_prompt = gr.Textbox(label="Negative Prompt", value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
1212
- with gr.Column():
1213
- result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery").style(grid=2,
1214
- height='auto')
1215
- ips = [input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode,
1216
- strength, scale, seed, eta, confidence, nms_thresh, condition_mode]
1217
- run_button.click(fn=process_bbox, inputs=ips, outputs=[result_gallery])
1218
 
1219
  with gr.TabItem("Outpainting"):
1220
  with gr.Row():
 
1185
  ddim_steps, guess_mode, strength, scale, seed, eta, condition_mode]
1186
  run_button.click(fn=process_seg, inputs=ips, outputs=[result_gallery])
1187
 
1188
+ # with gr.TabItem("Bbox"):
1189
+ # with gr.Row():
1190
+ # gr.Markdown("## UniControl Stable Diffusion with Object Bounding Boxes (MS-COCO)")
1191
+ # with gr.Row():
1192
+ # with gr.Column():
1193
+ # input_image = gr.Image(source='upload', type="numpy")
1194
+ # prompt = gr.Textbox(label="Prompt")
1195
+ # run_button = gr.Button(label="Run")
1196
+ # with gr.Accordion("Advanced options", open=False):
1197
+ # num_samples = gr.Slider(label="Images", minimum=1, maximum=12, value=1, step=1)
1198
+ # image_resolution = gr.Slider(label="Image Resolution", minimum=256, maximum=768, value=512,
1199
+ # step=64)
1200
+ # strength = gr.Slider(label="Control Strength", minimum=0.0, maximum=2.0, value=1.0, step=0.01)
1201
+ # condition_mode = gr.Checkbox(label='Condition Extraction: RGB -> Bbox', value=True)
1202
+ # guess_mode = gr.Checkbox(label='Guess Mode', value=False)
1203
+ # confidence = gr.Slider(label="Confidence of Detection", minimum=0.1, maximum=1.0, value=0.4,
1204
+ # step=0.1)
1205
+ # nms_thresh = gr.Slider(label="Nms Threshold", minimum=0.1, maximum=1.0, value=0.5, step=0.1)
1206
+ # ddim_steps = gr.Slider(label="Steps", minimum=1, maximum=100, value=20, step=1)
1207
+ # scale = gr.Slider(label="Guidance Scale", minimum=0.1, maximum=30.0, value=9.0, step=0.1)
1208
+ # seed = gr.Slider(label="Seed", minimum=-1, maximum=2147483647, step=1, randomize=True)
1209
+ # eta = gr.Number(label="eta (DDIM)", value=0.0)
1210
+ # a_prompt = gr.Textbox(label="Added Prompt", value='best quality, extremely detailed, bright')
1211
+ # n_prompt = gr.Textbox(label="Negative Prompt", value='longbody, lowres, bad anatomy, bad hands, missing fingers, extra digit, fewer digits, cropped, worst quality, low quality')
1212
+ # with gr.Column():
1213
+ # result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery").style(grid=2,
1214
+ # height='auto')
1215
+ # ips = [input_image, prompt, a_prompt, n_prompt, num_samples, image_resolution, ddim_steps, guess_mode,
1216
+ # strength, scale, seed, eta, confidence, nms_thresh, condition_mode]
1217
+ # run_button.click(fn=process_bbox, inputs=ips, outputs=[result_gallery])
1218
 
1219
  with gr.TabItem("Outpainting"):
1220
  with gr.Row():