fffiloni commited on
Commit
8eb93c7
·
1 Parent(s): 88d0e15

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -4
app.py CHANGED
@@ -20,10 +20,7 @@ device="cpu"
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(image_mode="RGB",
24
- source="upload",
25
- type="filepath")
26
-
27
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
28
 
29
  def infer(prompt, init_image):
 
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")
 
 
 
24
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
25
 
26
  def infer(prompt, init_image):