Vignesh455 commited on
Commit
32e5840
·
verified ·
1 Parent(s): 9dae3a4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -29,8 +29,8 @@ def predict(dict, prompt="", negative_prompt="", guidance_scale=7.5, steps=20, s
29
  scheduler = getattr(diffusers, scheduler_class_name)
30
  pipe.scheduler = scheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler", **add_kwargs)
31
 
32
- init_image = dict["image"].convert("RGB").resize((1024, 1024))
33
- mask = dict["mask"].convert("RGB").resize((1024, 1024))
34
 
35
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
36
 
 
29
  scheduler = getattr(diffusers, scheduler_class_name)
30
  pipe.scheduler = scheduler.from_pretrained("stabilityai/stable-diffusion-xl-base-1.0", subfolder="scheduler", **add_kwargs)
31
 
32
+ init_image = dict["image"].convert("RGB")
33
+ mask = dict["mask"].convert("RGB")
34
 
35
  output = pipe(prompt = prompt, negative_prompt=negative_prompt, image=init_image, mask_image=mask, guidance_scale=guidance_scale, num_inference_steps=int(steps), strength=strength)
36