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(); |