Update app.py

app.py

@@ -1,15 +1,27 @@
 import gradio as gr
 from diffusers import DiffusionPipeline
 
-pipeline = DiffusionPipeline.from_pretrained("dataautogpt3/OpenDalleV1.1")
+# Initialize the DiffusionPipeline with the specified model
+pipeline = DiffusionPipeline.from_pretrained("stablediffusionapi/juggernaut-xl-v5")
+pipeline.load_lora_weights("openskyml/dalle-3-xl")
 
+# Function to generate image based on the prompt
 def generate_image(prompt):
-    if prompt != '':
-        image = pipeline.generate(prompt=prompt)
-        return image
-    else:
-        return None
+    image = pipeline(prompt).images[0]
+    return image
 
-prompt_input = gr.Textbox(label="Enter your prompt here:")
+# Define the Gradio interface layout
+prompt_input = gr.textbox(label="", placeholder="Enter your prompt here...", lines=1, unresizeable=False)
+image_output = gr.Image(label="Generated Image")
 
-gr.Interface(generate_image, inputs=gr.Textbox(lines=5), outputs=gr.Image(label="Generated Image")).launch()
+# Create the Gradio interface with a more organized layout
+gr.Interface(
+    fn=generate_image,
+    inputs=gr.Textbox(lines=5, label="Enter Prompt", placeholder="Enter your prompt here..."),
+    outputs="image",
+    title="OpenJourney Image Generation",
+    theme="compact",  # Use a more compact theme for the interface
+    layout="vertical",  # Arrange the interface components vertically
+    inputs=[prompt_input],
+    outputs=image_output
+).launch()