Spaces:

AkiKagura
/

Marco-Generation

Runtime error

AkiKagura commited on Oct 31, 2022

Commit

c8e2317

•

1 Parent(s): 4716bf2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ pipe = StableDiffusionPipeline.from_pretrained("AkiKagura/mkgen-diffusion", use_
 pipe.safety_checker = empty_checker
 pipe.to(device)
-gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[2], height="auto")
 def infer(prompt):

 pipe.safety_checker = empty_checker
 pipe.to(device)
+gallery = gr.Gallery(label="Generated images", show_label=False, elem_id="gallery").style(grid=[1], height="auto")
 def infer(prompt):