mkrzyzan commited on
Commit
44fbbac
β€’
1 Parent(s): f863024

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -14,10 +14,10 @@ pipeline = pipeline.to("cuda")
14
  def predict(mask_img):
15
  prompt = "a green frog, highly detailed, natural lighting"
16
  image = pipeline(prompt=prompt,
17
- #num_inference_steps=35,
18
- image=mask_img["image"].convert("RGB").resize((512, 512)),
19
- mask_image=mask_img["mask"].convert("RGB").resize((512, 512))
20
- #guidance_scale=9
21
  ).images[0]
22
 
23
  return image
 
14
  def predict(mask_img):
15
  prompt = "a green frog, highly detailed, natural lighting"
16
  image = pipeline(prompt=prompt,
17
+ num_inference_steps=35,
18
+ image=mask_img["image"].convert("RGB"),
19
+ mask_image=mask_img["mask"].convert("RGB"),
20
+ guidance_scale=9
21
  ).images[0]
22
 
23
  return image