from diffusers import AutoPipelineForText2Image
import torch

def generateImage(text):
    pipeline = AutoPipelineForText2Image.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, variant="fp16")
    #pipeline.to("cuda")
    image = pipeline(text)
    return image.images[0]


gr.Interface(fn=generateImage, 
             inputs='textbox', 
             outputs='image').launch();