Hugging Face's logo

Spaces:
eshwar06
/
lipsmusemodel1
Runtime error

App Files Community
sk61tc4j53obiq commited on
Commit
d9b71e7
·
verified ·
1 Parent(s): bc248be

Fix zero gpu support

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -39,6 +39,7 @@ if os.getenv('SYSTEM') == 'spaces':
39
  ProjectDir = os.path.abspath(os.path.dirname(__file__))
40
  CheckpointsDir = os.path.join(ProjectDir, "models")
41
 
 
42
  @torch.no_grad()
43
  def debug_inpainting(video_path, bbox_shift, extra_margin=10, parsing_mode="jaw",
44
  left_cheek_width=90, right_cheek_width=90):
 
39
  ProjectDir = os.path.abspath(os.path.dirname(__file__))
40
  CheckpointsDir = os.path.join(ProjectDir, "models")
41
 
42
+ @spaces.GPU()
43
  @torch.no_grad()
44
  def debug_inpainting(video_path, bbox_shift, extra_margin=10, parsing_mode="jaw",
45
  left_cheek_width=90, right_cheek_width=90):