Update app.py

app.py

@@ -928,7 +928,7 @@ def generate_image_from_prompt(text_in, style_in):
 
 #loss_generated_img = (loss_style(prompt, style_embed[0], style_seed))
 
-	return [generated_image]
+	return [generated_image,generate_loss_details]
 
 
 # Define Interface
@@ -950,7 +950,7 @@ demo = gr.Interface(generate_image_from_prompt,
                         ),
                     ],
                     outputs=[
-                        gr.Gallery(label="Generated Images", show_label=False, elem_id="gallery", columns=1, rows=1,
+                        gr.Gallery(label="Generated Images", show_label=False, elem_id="gallery", columns=2, rows=2,
                                    object_fit="contain"),
                     ],
                     title=title,