File size: 271 Bytes
c24b225
5acc1ac
c24b225
5acc1ac
 
 
c24b225
5acc1ac
c24b225
1
2
3
4
5
6
7
8
9
import gradio as gr
from diffusers import DiffusionPipeline

pipeline = DiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
def generate(prompt):
    return pipeline(prompt)

iface = gr.Interface(fn=generate, inputs="text", outputs="image")
iface.launch()