Spaces:
Runtime error
Runtime error
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(); |