Abiru commited on
Commit
f06640e
1 Parent(s): e45c2ab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -4,9 +4,8 @@ import random
4
  import spaces
5
 
6
  pipe = DiffusionPipeline.from_pretrained("UnfilteredAI/NSFW-gen-v2")
7
- pipe.to('cuda')
8
 
9
- @spaces.GPU
10
  def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):
11
  return pipe(
12
  prompt=prompt,
 
4
  import spaces
5
 
6
  pipe = DiffusionPipeline.from_pretrained("UnfilteredAI/NSFW-gen-v2")
 
7
 
8
+
9
  def generate(prompt, negative_prompt, num_inference_steps, guidance_scale, width, height, num_samples):
10
  return pipe(
11
  prompt=prompt,