Spaces:

rlawjdghek
/

StableVITON

Running on Zero

rlawjdghek commited on Mar 30

Commit

644b248

•

1 Parent(s): 22eed09

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,4 +1,4 @@
-# import spaces  # TODO: turn on when final upload
 import os
 import sys
 import time
@@ -87,7 +87,7 @@ def stable_viton_model_hd(
     pil_output = Image.fromarray(output)
     return pil_output
-# @spaces.GPU  # TODO: turn on when final upload
 @torch.no_grad()
 def process_hd(vton_img, garm_img, n_steps):
     model_type = 'hd'

+import spaces
 import os
 import sys
 import time
     pil_output = Image.fromarray(output)
     return pil_output
+@spaces.GPU  # TODO: turn on when final upload
 @torch.no_grad()
 def process_hd(vton_img, garm_img, n_steps):
     model_type = 'hd'