rev all

app.py

@@ -40,8 +40,7 @@ def infer(prompt, negative_prompt, guidance_scale=10, num_inference_steps=50):
             
     return imgs
     
-# output = gr.Gallery(label="Outputs").style(grid=(1,2))
-output = gr.Gallery(label="Outputs", rows=1, columns=4, object_fit="contain", height="auto")#, columns=2, rows=1)
+output = gr.Gallery(label="Outputs").style(grid=(1,2))
 
 # customize interface
 title = "KerasCV Stable Diffusion Demo on images of Ace."
@@ -54,8 +53,8 @@ examples=[
     ["Photo of cute puggieace dog as super hero, futuristic city in background, cinematic light, high dynamic range, insane intricate details, stunning cinema effects, aesthetic, masterpiece, trending on artstation, cartoon art", "", 12, 75],
 ]
 
-base_14 = "https://huggingface.co/nielsgl/dreambooth-pug-ace-sd1.4/tree/main/"
-base_21 = "https://huggingface.co/nielsgl/dreambooth-keras-pug-ace-sd2.1/tree/main/"
+base_14 = "https://huggingface.co/nielsgl/dreambooth-pug-ace-sd1.4/resolve/main/"
+base_21 = "https://huggingface.co/nielsgl/dreambooth-keras-pug-ace-sd2.1/resolve/main/"
 
 model_card_1 = f"""
 # KerasCV Stable Diffusion in Diffusers 🧨🤗
@@ -176,7 +175,7 @@ with gr.Blocks() as demo:
             prompt_steps = gr.Slider(label="Inference Steps",value=50)
             prompt_btn = gr.Button("Generate")
         with gr.Column():
-            output = gr.Gallery(label="Outputs", object_fit="contain", height="auto")#.style(grid=(1,2))
+            output = gr.Gallery(label="Outputs").style(grid=(1,2))
     prompt_btn.click(infer, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=[output])
     with gr.Row():
         gr.Examples(examples, inputs=[prompt_pos, prompt_neg, prompt_gs, prompt_steps], outputs=output, fn=infer, cache_examples=True)
@@ -192,12 +191,6 @@ with gr.Blocks() as demo:
         with gr.Column():
             gr.Markdown(" ")
 
-    with gr.Row():
-        with gr.Column():
-            gr.Markdown("""# Hola
-            ![blabla](https://huggingface.co/nielsgl/dreambooth-pug-ace-sd1.4/resolve/main/examples/emperor-1.4.jpeg)
-            """)
-
 demo.queue().launch()
 
 
@@ -213,4 +206,4 @@ demo.queue().launch()
 #         ], outputs=[output], title=title, description=description, examples=examples).queue()
 
 
-# gr.TabbedInterface([card_interface, demo_interface], ["Model Card", "Demo 🤗"]).launch()
+# gr.TabbedInterface([card_interface, demo_interface], ["Model Card", "Demo 🤗"]).launch()