Update app.py

app.py

@@ -1,4 +1,5 @@
 import gradio as gr
+from sd_utils import *
 
 
 
@@ -28,12 +29,12 @@ with gr.Blocks() as interface:
     with gr.Row():
         outputs = gr.Gallery(
             label="Generated Image", show_label=True,
-            columns=[2], rows=[1], object_fit="contain"
+            columns=[1], rows=[1], object_fit="contain"
         )
 
     with gr.Row():
         button = gr.Button("Generate")
-        button.click(generate_art, inputs=inputs, outputs=outputs)
+        button.click(generate_with_embs_custom_loss, inputs=inputs, outputs=outputs)
 
     with gr.Row():
         gr.Examples(examples=get_examples(), inputs=inputs, outputs=outputs, fn=generate_art, cache_examples=True)