Spaces:
Runtime error
Runtime error
File size: 488 Bytes
97cf2c4 523cfbe 97cf2c4 589c3a2 523cfbe 589c3a2 523cfbe 7cba8bf 523cfbe |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
from diffusers import StableDiffusionPipeline
text2img = StableDiffusionPipeline.from_pretrained("Mitsua/mitsua-diffusion-one")
text2img.safety_checker=None # It is practically impossible for Mitsua to generate NSFW images
text2img.to("cuda")
def predict(text):
image = text2img(text).images[0]
return image
demo = gr.Interface(
fn=predict,
inputs='text',
outputs='image',
)
demo.launch()
#gr.Interface.load("models/Mitsua/mitsua-diffusion-one").launch() |