Spaces:

Organika
/

mitsua-diffusion-one

Runtime error

umm-maybe commited on Nov 28, 2023

Commit

7cba8bf

1 Parent(s): 523cfbe

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -5,7 +5,8 @@ text2img = StableDiffusionPipeline.from_pretrained("Mitsua/mitsua-diffusion-one"
 text2img.safety_checker=None # It is practically impossible for Mitsua to generate NSFW images
 def predict(text):
-  return image = text2img(text).images[0]
 demo = gr.Interface(
   fn=predict,

 text2img.safety_checker=None # It is practically impossible for Mitsua to generate NSFW images
 def predict(text):
+  image = text2img(text).images[0]
+  return image
 demo = gr.Interface(
   fn=predict,