AkiKagura commited on
Commit
b61c69d
·
1 Parent(s): 435ad84

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ device="cpu"
21
  img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("AkiKagura/mkgen-diffusion", use_auth_token=YOUR_TOKEN)
22
  img_pipe.to(device)
23
 
24
- source_img = gr.Image(source="canvas", type="filepath", tool='color-sketch', label="new gradio color sketch")
25
 
26
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[1], height="auto")
27
 
 
21
  img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("AkiKagura/mkgen-diffusion", use_auth_token=YOUR_TOKEN)
22
  img_pipe.to(device)
23
 
24
+ source_img = gr.Image(source="canvas", type="filepath", tool='color-sketch', shape=[512, 512], label="new gradio color sketch")
25
 
26
  gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[1], height="auto")
27