renyuxi commited on
Commit
467bbf7
·
verified ·
1 Parent(s): 07c50cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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.,