Spaces:
Running
Running
Update app.py
Browse files
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("
|
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
|