FrankFacundo commited on
Commit
d76eab0
·
1 Parent(s): d2f86ce
Files changed (1) hide show
  1. app.py +14 -16
app.py CHANGED
@@ -306,17 +306,16 @@ def infer(
306
  # --------------------------------------------------
307
  css = """#wrapper {max-width: 960px; margin: 0 auto;}"""
308
  with gr.Blocks(css=css, elem_id="wrapper") as demo:
309
- gr.Markdown("## FLUX.1‑dev‑ControlNet‑Union‑Pro")
310
  gr.Markdown(
311
  "A unified ControlNet for **FLUX.1‑dev** from the InstantX team and Shakker Labs. "
312
- + "Recommended strengths: *canny 0.65*, *tile 0.45*, *depth 0.55*, *blur 0.45*, "
313
- + "*openpose 0.55*, *gray 0.45*, *low quality 0.40*. Long prompts usually help."
314
  )
315
 
316
  # ------------ Image panel row ------------
317
  with gr.Row():
318
  control_image = gr.Image(
319
- label="Upload a processed control image",
320
  type="pil",
321
  height=512 + 256,
322
  )
@@ -324,7 +323,7 @@ with gr.Blocks(css=css, elem_id="wrapper") as demo:
324
  preview_image = gr.Image(label="Pre‑processed Cond", height=512 + 256)
325
 
326
  # ------------ Prompt ------------
327
- prompt_txt = gr.Textbox(label="Prompt", value="Una imagen bonita", lines=1)
328
 
329
  # ------------ ControlNet settings ------------
330
  with gr.Row():
@@ -334,18 +333,8 @@ with gr.Blocks(css=css, elem_id="wrapper") as demo:
334
  choices=list(MODE_MAPPING.keys()), value="canny", label="Mode"
335
  )
336
  strength_slider = gr.Slider(
337
- 0.0, 1.0, value=0.9, step=0.01, label="control strength"
338
  )
339
- with gr.Column():
340
- seed_slider = gr.Slider(0, MAX_SEED, step=1, value=42, label="Seed")
341
- randomize_chk = gr.Checkbox(label="Randomize seed", value=False)
342
- guidance_slider = gr.Slider(
343
- 0.0, 10.0, step=0.1, value=3.5, label="Guidance scale"
344
- )
345
- steps_slider = gr.Slider(1, 50, step=1, value=24, label="Inference steps")
346
-
347
- with gr.Row():
348
- with gr.Column():
349
  gr.Markdown("### Preprocess")
350
  canny_threshold_1 = gr.Slider(
351
  0, 500, step=1, value=100, label="Canny threshold 1"
@@ -353,6 +342,15 @@ with gr.Blocks(css=css, elem_id="wrapper") as demo:
353
  canny_threshold_2 = gr.Slider(
354
  0, 500, step=1, value=200, label="Canny threshold 2"
355
  )
 
 
 
 
 
 
 
 
 
356
  submit_btn = gr.Button("Submit")
357
 
358
  submit_btn.click(
 
306
  # --------------------------------------------------
307
  css = """#wrapper {max-width: 960px; margin: 0 auto;}"""
308
  with gr.Blocks(css=css, elem_id="wrapper") as demo:
309
+ gr.Markdown("## FLUX.1‑dev‑ControlNet‑Union‑Pro by Frank")
310
  gr.Markdown(
311
  "A unified ControlNet for **FLUX.1‑dev** from the InstantX team and Shakker Labs. "
312
+ + "Recommended strengths: *canny 0.76*. Long prompts usually help."
 
313
  )
314
 
315
  # ------------ Image panel row ------------
316
  with gr.Row():
317
  control_image = gr.Image(
318
+ label="Upload animage",
319
  type="pil",
320
  height=512 + 256,
321
  )
 
323
  preview_image = gr.Image(label="Pre‑processed Cond", height=512 + 256)
324
 
325
  # ------------ Prompt ------------
326
+ prompt_txt = gr.Textbox(label="Prompt", value="White background", lines=1)
327
 
328
  # ------------ ControlNet settings ------------
329
  with gr.Row():
 
333
  choices=list(MODE_MAPPING.keys()), value="canny", label="Mode"
334
  )
335
  strength_slider = gr.Slider(
336
+ 0.0, 1.0, value=0.76, step=0.01, label="control strength"
337
  )
 
 
 
 
 
 
 
 
 
 
338
  gr.Markdown("### Preprocess")
339
  canny_threshold_1 = gr.Slider(
340
  0, 500, step=1, value=100, label="Canny threshold 1"
 
342
  canny_threshold_2 = gr.Slider(
343
  0, 500, step=1, value=200, label="Canny threshold 2"
344
  )
345
+
346
+ with gr.Column():
347
+ seed_slider = gr.Slider(0, MAX_SEED, step=1, value=42, label="Seed")
348
+ randomize_chk = gr.Checkbox(label="Randomize seed", value=False)
349
+ guidance_slider = gr.Slider(
350
+ 0.0, 10.0, step=0.1, value=3.5, label="Guidance scale"
351
+ )
352
+ steps_slider = gr.Slider(1, 50, step=1, value=50, label="Inference steps")
353
+
354
  submit_btn = gr.Button("Submit")
355
 
356
  submit_btn.click(