Hugging Face's logo

Spaces:
fffiloni
/
stable-diffusion-img2img
Running

App Files Community
40
fffiloni commited on
Commit
98e0ddb
·
1 Parent(s): 5665597

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ device="cpu"
19
 
20
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
21
  pipe.to(device)
22
-
23
  source_img = gr.Image(source="upload", type="pil", label="init_img", shape=(image_size, image_size))
24
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
25
 
 
19
 
20
  pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
21
  pipe.to(device)
22
+ image_size = 512
23
  source_img = gr.Image(source="upload", type="pil", label="init_img", shape=(image_size, image_size))
24
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
25