comrender commited on
Commit
c5f42d3
·
verified ·
1 Parent(s): a4b4b11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -31
app.py CHANGED
@@ -171,11 +171,9 @@ def enhance_image(
171
  image_url,
172
  seed,
173
  randomize_seed,
174
- num_inference_steps,
175
  upscale_factor,
176
  denoising_strength,
177
  custom_prompt,
178
- tile_size,
179
  progress=gr.Progress(track_tqdm=True),
180
  ):
181
  if image_input is not None:
@@ -249,14 +247,14 @@ def enhance_image(
249
  usd_res = ultimatesdupscale.upscale(
250
  upscale_by=upscale_factor,
251
  seed=seed,
252
- steps=num_inference_steps,
253
  cfg=1,
254
  sampler_name="euler",
255
  scheduler="normal",
256
  denoise=denoising_strength,
257
  mode_type="Linear",
258
- tile_width=tile_size,
259
- tile_height=tile_size,
260
  mask_blur=8,
261
  tile_padding=32,
262
  seam_fix_mode="None",
@@ -339,15 +337,6 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX ComfyUI") as demo:
339
  info="How much to upscale the image"
340
  )
341
 
342
- num_inference_steps = gr.Slider(
343
- label="Number of Inference Steps",
344
- minimum=1,
345
- maximum=50,
346
- step=1,
347
- value=25,
348
- info="More steps = better quality but slower"
349
- )
350
-
351
  denoising_strength = gr.Slider(
352
  label="Denoising Strength",
353
  minimum=0.0,
@@ -356,27 +345,16 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX ComfyUI") as demo:
356
  value=0.3,
357
  info="Controls how much the image is transformed"
358
  )
359
-
360
- tile_size = gr.Slider(
361
- label="Tile Size",
362
- minimum=256,
363
- maximum=2048,
364
- step=64,
365
- value=1024,
366
- info="Size of tiles for processing (larger = faster but more memory)"
367
- )
368
 
369
  with gr.Row():
370
  randomize_seed = gr.Checkbox(
371
  label="Randomize seed",
372
  value=True
373
  )
374
- seed = gr.Slider(
375
  label="Seed",
376
- minimum=0,
377
- maximum=MAX_SEED,
378
- step=1,
379
- value=42,
380
  interactive=True
381
  )
382
 
@@ -404,11 +382,9 @@ with gr.Blocks(css=css, title="🎨 AI Image Upscaler - FLUX ComfyUI") as demo:
404
  image_url,
405
  seed,
406
  randomize_seed,
407
- num_inference_steps,
408
  upscale_factor,
409
  denoising_strength,
410
- custom_prompt,
411
- tile_size
412
  ],
413
  outputs=[result_slider]
414
  )
 
171
  image_url,
172
  seed,
173
  randomize_seed,
 
174
  upscale_factor,
175
  denoising_strength,
176
  custom_prompt,
 
177
  progress=gr.Progress(track_tqdm=True),
178
  ):
179
  if image_input is not None:
 
247
  usd_res = ultimatesdupscale.upscale(
248
  upscale_by=upscale_factor,
249
  seed=seed,
250
+ steps=25,
251
  cfg=1,
252
  sampler_name="euler",
253
  scheduler="normal",
254
  denoise=denoising_strength,
255
  mode_type="Linear",
256
+ tile_width=1024,
257
+ tile_height=1024,
258
  mask_blur=8,
259
  tile_padding=32,
260
  seam_fix_mode="None",
 
337
  info="How much to upscale the image"
338
  )
339
 
 
 
 
 
 
 
 
 
 
340
  denoising_strength = gr.Slider(
341
  label="Denoising Strength",
342
  minimum=0.0,
 
345
  value=0.3,
346
  info="Controls how much the image is transformed"
347
  )
 
 
 
 
 
 
 
 
 
348
 
349
  with gr.Row():
350
  randomize_seed = gr.Checkbox(
351
  label="Randomize seed",
352
  value=True
353
  )
354
+ seed = gr.Textbox(
355
  label="Seed",
356
+ value="42",
357
+ placeholder="Enter seed value",
 
 
358
  interactive=True
359
  )
360
 
 
382
  image_url,
383
  seed,
384
  randomize_seed,
 
385
  upscale_factor,
386
  denoising_strength,
387
+ custom_prompt
 
388
  ],
389
  outputs=[result_slider]
390
  )