SkalskiP commited on
Commit
8a84578
1 Parent(s): 4fda942
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -79,7 +79,7 @@ def process(
79
  if randomize_seed_checkbox:
80
  seed_slicer = random.randint(0, MAX_SEED)
81
  generator = torch.Generator().manual_seed(seed_slicer)
82
- return pipe(
83
  prompt=input_text,
84
  image=resized_image,
85
  mask_image=resized_mask,
@@ -88,7 +88,9 @@ def process(
88
  strength=strength_slider,
89
  generator=generator,
90
  num_inference_steps=num_inference_steps_slider
91
- ).images[0], resized_mask
 
 
92
 
93
 
94
  with gr.Blocks() as demo:
 
79
  if randomize_seed_checkbox:
80
  seed_slicer = random.randint(0, MAX_SEED)
81
  generator = torch.Generator().manual_seed(seed_slicer)
82
+ result = pipe(
83
  prompt=input_text,
84
  image=resized_image,
85
  mask_image=resized_mask,
 
88
  strength=strength_slider,
89
  generator=generator,
90
  num_inference_steps=num_inference_steps_slider
91
+ ).images[0]
92
+ print('INFERENCE DONE')
93
+ return result, resized_mask
94
 
95
 
96
  with gr.Blocks() as demo: