Spaces:
Runtime error
Runtime error
import gradio as gr | |
from diffusers import StableDiffusionPipeline, EulerDiscreteScheduler | |
import torch | |
def txt2img(prompt): | |
image = pipe(prompt, height=768, width=768, guidance_scale = 10).images[0] | |
# image.save("sd_image.png") | |
return image | |
gr.Interface(txt2img, gr.Text(), gr.Image(), title = 'Stable Diffusion 2.0 Colab with Gradio UI').launch(share = True, debug = True) |