Update app.py

app.py

@@ -27,13 +27,18 @@ def generate_image_from_model(prompt):
 
     return image_path  # Return the path to the image
 
-# Create the Gradio interface
+# Define the chatbot function to return the generated image
+def chatbot(prompt):
+    image = generate_image(prompt)
+    return image
+
+# Create the Gradio interface with the same UI/UX
 interface = gr.Interface(
-    fn=generate_image_from_model,
+    fn=chatbot,
     inputs="text",
     outputs="image",
-    title="OpenJourney Text-to-Image",
-    description="Enter a prompt to generate an image using OpenJourney model."
+    title="prompthero/openjourney",
+    description="Enter a text prompt and get an AI-generated image."
 )
 
 # Launch the interface