Update app.py

app.py

@@ -1,3 +1,18 @@
 import gradio as gr
+import spaces
+from diffusers import DiffusionPipeline
 
-gr.load("models/TheMistoAI/MistoLine").launch()
+
+model_name = 'UnfilteredAI/NSFW-gen-v2'
+pipe = DiffusionPipeline.from_pretrained(model_name)
+pipe.to('cuda')
+
+@spaces.GPU
+def generate(prompt):
+    return pipe(prompt).images
+
+gr.Interface(
+    fn=generate,
+    inputs=gr.Text(),
+    outputs=gr.Gallery(),
+).launch()