Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -59,7 +59,7 @@ with gr.Blocks() as demo:
|
|
59 |
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.float16), timer("inference"):
|
60 |
return pipe(
|
61 |
prompt=[prompt]*num_images,
|
62 |
-
image=[scribble.resize((512, 512))]*num_images,
|
63 |
generator=torch.Generator().manual_seed(int(seed)),
|
64 |
num_inference_steps=steps,
|
65 |
guidance_scale=0.,
|
|
|
59 |
with torch.inference_mode(), torch.autocast("cuda", dtype=torch.float16), timer("inference"):
|
60 |
return pipe(
|
61 |
prompt=[prompt]*num_images,
|
62 |
+
image=[scribble['composite'].resize((512, 512))]*num_images,
|
63 |
generator=torch.Generator().manual_seed(int(seed)),
|
64 |
num_inference_steps=steps,
|
65 |
guidance_scale=0.,
|