import gradio as gr
import torch
from diffusers import StableDiffusionPipeline

pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", torch_dtype=torch.float16)

def diffusion(prompt, filename):
    image = pipe(prompt).images[0]
    image.save(filename)
    return image

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