mkrzyzan commited on
Commit
84bc0a5
β€’
1 Parent(s): 4666901

Update app.py

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