Abiru commited on
Commit
45714ae
1 Parent(s): e2ee1af

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from diffusers import DiffusionPipeline
3
  import random
4
 
5
- pipe = DiffusionPipeline.from_pretrained("models/UnfilteredAI/NSFW-gen-v2")
6
 
7
  def generate_image(prompt, negative_prompt="", height=1024, width=1024, seed=None):
8
  generator = None
 
2
  from diffusers import DiffusionPipeline
3
  import random
4
 
5
+ pipe = DiffusionPipeline.from_pretrained("UnfilteredAI/NSFW-gen-v2")
6
 
7
  def generate_image(prompt, negative_prompt="", height=1024, width=1024, seed=None):
8
  generator = None