debug
Browse files
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 |
-
|
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]
|
|
|
|
|
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:
|