ShreeKanade07 commited on
Commit
4de503a
·
1 Parent(s): ad6741c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def predict(image,mask,prompt):
34
  strength=0.9
35
  generator = torch.manual_seed(32)
36
  negative_prompt="zoomed in, blurry, oversaturated, warped,artifacts,flickers"
37
- images = pipe(prompt=prompt, image=image, mask_image=mask_image, strength=strength, negative_prompt=negative_prompt, generator=generator).images
38
  return images[0]
39
 
40
 
 
34
  strength=0.9
35
  generator = torch.manual_seed(32)
36
  negative_prompt="zoomed in, blurry, oversaturated, warped,artifacts,flickers"
37
+ images = pipe(prompt=prompt, image=image, mask_image=mask_image, strength=strength, negative_prompt=negative_prompt, generator=generator,num_inference_steps=20).images
38
  return images[0]
39
 
40