File size: 568 Bytes
4ec204b ffde86b 4ec204b ffde86b 4ec204b ffde86b 4ec204b ffde86b 4ec204b ffde86b | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import gradio as gr
from diffusers import StableDiffusionPipeline
# Load Stable Diffusion Model
model = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v-1-4-original")
# Function to generate images
def generate_image(prompt):
image = model(prompt).images[0]
return image
# Create Gradio interface
iface = gr.Interface(fn=generate_image,
inputs=gr.Textbox(label="Enter a prompt"),
outputs=gr.Image(label="Generated Image"),
live=True)
# Launch the interface
iface.launch()
|