Upload 2 files

app.py

@@ -0,0 +1,13 @@
+import gradio as gr
+from diffusers import StableDiffusionPipeline
+
+# Load your model (make sure it's downloaded from Hugging Face)
+pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v-1-4-original")
+
+# Define the function to generate the image
+def generate_image(prompt):
+    image = pipe(prompt).images[0]
+    return image
+
+# Set up Gradio interface
+gr.Interface(fn=generate_image, inputs="text", outputs="image").launch()

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+diffusers
+torch
+transformers