Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
|