Himanshu-AT commited on
Commit
4e648a1
·
1 Parent(s): a9d7fe1
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -89,7 +89,7 @@ def infer(edit_images, prompt, seed=42, randomize_seed=False, width=1024, height
89
  if randomize_seed:
90
  seed = random.randint(0, MAX_SEED)
91
 
92
- controlImage = processor(image)[0].convert("RGB")
93
  image = pipe(
94
  mask_image_latent=vae.encode(controlImage),
95
  prompt=prompt,
 
89
  if randomize_seed:
90
  seed = random.randint(0, MAX_SEED)
91
 
92
+ controlImage = processor(image)
93
  image = pipe(
94
  mask_image_latent=vae.encode(controlImage),
95
  prompt=prompt,