Spaces:
Running
on
Zero
Running
on
Zero
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:
|