Boboiazumi commited on
Commit
c32783f
·
verified ·
1 Parent(s): 1110322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import os
2
  import time
3
  import shutil
@@ -26,7 +27,7 @@ def _handle_roop_solve(
26
 
27
  return message, ROOP_OUTPUT_VIDEO_PATH
28
 
29
-
30
  def handle_roop(
31
  video_path: str,
32
  img_path: str
 
1
+ import spaces
2
  import os
3
  import time
4
  import shutil
 
27
 
28
  return message, ROOP_OUTPUT_VIDEO_PATH
29
 
30
+ @spaces.GPU
31
  def handle_roop(
32
  video_path: str,
33
  img_path: str