Create app.py

app.py

@@ -0,0 +1,20 @@
+from diffusers import DiffusionPipeline
+from gradio import Gradio
+
+# Load the pre-trained model
+pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+
+# Create a Gradio interface
+interface = Gradio(
+    pipeline,
+    input_prompt="Enter a prompt for the image generation model:",
+    output_prompt="Generated image:",
+)
+
+# Generate the image
+image = interface(prompt)
+
+# Display the generated image
+import matplotlib.pyplot as plt
+plt.imshow(image)
+plt.show()