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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ 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"
 
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"