Hugging Face's logo

Spaces:
Ariamehr
/
Meta-Sapiens
Runtime error

App Files Community
2
Ariamehr commited on
Commit
860b441
·
verified ·
1 Parent(s): 8c46161

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ from diffusers import DiffusionPipeline
16
  pipe = DiffusionPipeline.from_pretrained(...)
17
  pipe.to('cuda')
18
 
19
- +@spaces.GPU
20
  def generate(prompt):
21
  return pipe(prompt).images
22
 
 
16
  pipe = DiffusionPipeline.from_pretrained(...)
17
  pipe.to('cuda')
18
 
19
+ @spaces.GPU
20
  def generate(prompt):
21
  return pipe(prompt).images
22