txt2image / app.py
rsadaphule's picture
Create app.py
b343fac verified
raw
history blame contribute delete
No virus
405 Bytes
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();