Update app.py

app.py

@@ -1068,4 +1068,34 @@ 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]
+
+
+# Define Interface
+title = 'Stable Diffusion Art Generator'
+
+# Add clear and concise labels and instructions
+prompt_label = "Enter a prompt (e.g., 'A campfire (oil on canvas)'"
+styles_label = "Select a Pretrained Style:"
+
+instructions = "Explore creative art generation using Stable Diffusion. Enter a prompt and choose a style to get started."
+
+demo = gr.Interface(generate_image_from_prompt,
+                    inputs=[
+                        gr.Textbox('A campfire (oil on canvas)', label=prompt_label),
+                        gr.Dropdown(
+                            ['learned_embeds_gartic-phone', 'learned_embeds_hawaiian-shirt', 'learned_embeds_phone01', 'learned_embeds_style-spdmn', 'learned_embedssd_yvmqznrm'],
+                            value="learned_embeds_gartic-phone",
+                            label=styles_label
+                        ),
+                    ],
+                    outputs=[
+                        gr.Gallery(label="Generated Images", show_label=False, elem_id="gallery", columns=[2], rows=[2],
+                                   object_fit="contain"),
+                    ],
+                    title=title,
+                    description=instructions
+                   )
+
+demo.launch(debug=True)
+