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