lzyhha commited on
Commit
4daee5e
·
1 Parent(s): 4011725
Files changed (1) hide show
  1. app.py +16 -32
app.py CHANGED
@@ -384,113 +384,97 @@ def create_demo(model):
384
  partial(process_tasks, func=demo_tasks.process_dense_prediction_tasks),
385
  inputs=[dense_prediction_tasks],
386
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
387
- show_progress="full",
388
- queue=False)
389
 
390
  conditional_generation_tasks.click(
391
  partial(process_tasks, func=demo_tasks.process_conditional_generation_tasks),
392
  inputs=[conditional_generation_tasks],
393
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
394
- show_progress="full",
395
- queue=False)
396
 
397
  image_restoration_tasks.click(
398
  partial(process_tasks, func=demo_tasks.process_image_restoration_tasks),
399
  inputs=[image_restoration_tasks],
400
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
401
- show_progress="full",
402
- queue=False)
403
 
404
  style_transfer_tasks.click(
405
  partial(process_tasks, func=demo_tasks.process_style_transfer_tasks),
406
  inputs=[style_transfer_tasks],
407
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
408
- show_progress="full",
409
- queue=False)
410
 
411
  style_condition_fusion_tasks.click(
412
  partial(process_tasks, func=demo_tasks.process_style_condition_fusion_tasks),
413
  inputs=[style_condition_fusion_tasks],
414
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
415
- show_progress="full",
416
- queue=False)
417
 
418
  relighting_tasks.click(
419
  partial(process_tasks, func=demo_tasks.process_relighting_tasks),
420
  inputs=[relighting_tasks],
421
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
422
- show_progress="full",
423
- queue=False)
424
 
425
  tryon_tasks.click(
426
  partial(process_tasks, func=demo_tasks.process_tryon_tasks),
427
  inputs=[tryon_tasks],
428
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
429
- show_progress="full",
430
- queue=False)
431
 
432
  photodoodle_tasks.click(
433
  partial(process_tasks, func=demo_tasks.process_photodoodle_tasks),
434
  inputs=[photodoodle_tasks],
435
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
436
- show_progress="full",
437
- queue=False)
438
 
439
  editing_tasks.click(
440
  partial(process_tasks, func=demo_tasks.process_editing_tasks),
441
  inputs=[editing_tasks],
442
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
443
- show_progress="full",
444
- queue=False)
445
 
446
  unseen_tasks.click(
447
  partial(process_tasks, func=demo_tasks.process_unseen_tasks),
448
  inputs=[unseen_tasks],
449
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
450
- show_progress="full",
451
- queue=False)
452
 
453
  subject_driven_tasks.click(
454
  partial(process_tasks, func=demo_tasks.process_subject_driven_tasks),
455
  inputs=[subject_driven_tasks],
456
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
457
- show_progress="full",
458
- queue=False)
459
 
460
  style_transfer_with_subject_tasks.click(
461
  partial(process_tasks, func=demo_tasks.process_style_transfer_with_subject_tasks),
462
  inputs=[style_transfer_with_subject_tasks],
463
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
464
- show_progress="full",
465
- queue=False)
466
 
467
  condition_subject_fusion_tasks.click(
468
  partial(process_tasks, func=demo_tasks.process_condition_subject_fusion_tasks),
469
  inputs=[condition_subject_fusion_tasks],
470
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
471
- show_progress="full",
472
- queue=False)
473
 
474
  condition_subject_style_fusion_tasks.click(
475
  partial(process_tasks, func=demo_tasks.process_condition_subject_style_fusion_tasks),
476
  inputs=[condition_subject_style_fusion_tasks],
477
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
478
- show_progress="full",
479
- queue=False)
480
 
481
  editing_with_subject_tasks.click(
482
  partial(process_tasks, func=demo_tasks.process_editing_with_subject_tasks),
483
  inputs=[editing_with_subject_tasks],
484
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
485
- show_progress="full",
486
- queue=False)
487
 
488
  image_restoration_with_subject_tasks.click(
489
  partial(process_tasks, func=demo_tasks.process_image_restoration_with_subject_tasks),
490
  inputs=[image_restoration_with_subject_tasks],
491
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
492
- show_progress="full",
493
- queue=False)
494
  # Initialize grid
495
  model.set_grid_size(default_grid_h, default_grid_w)
496
 
 
384
  partial(process_tasks, func=demo_tasks.process_dense_prediction_tasks),
385
  inputs=[dense_prediction_tasks],
386
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
387
+ show_progress="full")
 
388
 
389
  conditional_generation_tasks.click(
390
  partial(process_tasks, func=demo_tasks.process_conditional_generation_tasks),
391
  inputs=[conditional_generation_tasks],
392
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
393
+ show_progress="full")
 
394
 
395
  image_restoration_tasks.click(
396
  partial(process_tasks, func=demo_tasks.process_image_restoration_tasks),
397
  inputs=[image_restoration_tasks],
398
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
399
+ show_progress="full")
 
400
 
401
  style_transfer_tasks.click(
402
  partial(process_tasks, func=demo_tasks.process_style_transfer_tasks),
403
  inputs=[style_transfer_tasks],
404
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
405
+ show_progress="full")
 
406
 
407
  style_condition_fusion_tasks.click(
408
  partial(process_tasks, func=demo_tasks.process_style_condition_fusion_tasks),
409
  inputs=[style_condition_fusion_tasks],
410
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
411
+ show_progress="full")
 
412
 
413
  relighting_tasks.click(
414
  partial(process_tasks, func=demo_tasks.process_relighting_tasks),
415
  inputs=[relighting_tasks],
416
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
417
+ show_progress="full")
 
418
 
419
  tryon_tasks.click(
420
  partial(process_tasks, func=demo_tasks.process_tryon_tasks),
421
  inputs=[tryon_tasks],
422
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
423
+ show_progress="full")
 
424
 
425
  photodoodle_tasks.click(
426
  partial(process_tasks, func=demo_tasks.process_photodoodle_tasks),
427
  inputs=[photodoodle_tasks],
428
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
429
+ show_progress="full")
 
430
 
431
  editing_tasks.click(
432
  partial(process_tasks, func=demo_tasks.process_editing_tasks),
433
  inputs=[editing_tasks],
434
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
435
+ show_progress="full")
 
436
 
437
  unseen_tasks.click(
438
  partial(process_tasks, func=demo_tasks.process_unseen_tasks),
439
  inputs=[unseen_tasks],
440
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
441
+ show_progress="full")
 
442
 
443
  subject_driven_tasks.click(
444
  partial(process_tasks, func=demo_tasks.process_subject_driven_tasks),
445
  inputs=[subject_driven_tasks],
446
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
447
+ show_progress="full")
 
448
 
449
  style_transfer_with_subject_tasks.click(
450
  partial(process_tasks, func=demo_tasks.process_style_transfer_with_subject_tasks),
451
  inputs=[style_transfer_with_subject_tasks],
452
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
453
+ show_progress="full")
 
454
 
455
  condition_subject_fusion_tasks.click(
456
  partial(process_tasks, func=demo_tasks.process_condition_subject_fusion_tasks),
457
  inputs=[condition_subject_fusion_tasks],
458
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
459
+ show_progress="full")
 
460
 
461
  condition_subject_style_fusion_tasks.click(
462
  partial(process_tasks, func=demo_tasks.process_condition_subject_style_fusion_tasks),
463
  inputs=[condition_subject_style_fusion_tasks],
464
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
465
+ show_progress="full")
 
466
 
467
  editing_with_subject_tasks.click(
468
  partial(process_tasks, func=demo_tasks.process_editing_with_subject_tasks),
469
  inputs=[editing_with_subject_tasks],
470
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
471
+ show_progress="full")
 
472
 
473
  image_restoration_with_subject_tasks.click(
474
  partial(process_tasks, func=demo_tasks.process_image_restoration_with_subject_tasks),
475
  inputs=[image_restoration_with_subject_tasks],
476
  outputs=[output_gallery] + all_image_inputs + [grid_h, grid_w, layout_prompt, task_prompt, content_prompt, upsampling_noise, steps],
477
+ show_progress="full")
 
478
  # Initialize grid
479
  model.set_grid_size(default_grid_h, default_grid_w)
480