Boboiazumi commited on
Commit
e09ea8e
·
verified ·
1 Parent(s): fee6538

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -20,14 +20,14 @@ def _handle_roop_solve(
20
  if not os.path.exists("media"):
21
  os.mkdir("media")
22
  start_time = time.time()
23
- command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_VIDEO_PATH} --execution-threads 512"
24
  os.system(command)
25
  solved_time = time.time() - start_time
26
  message = "Successfully solve the GED problem, using time ({:.3f}s).".format(solved_time)
27
 
28
  return message, ROOP_OUTPUT_VIDEO_PATH
29
 
30
- @spaces.GPU(duration=600)
31
  def handle_roop(
32
  video_path: str,
33
  img_path: str
 
20
  if not os.path.exists("media"):
21
  os.mkdir("media")
22
  start_time = time.time()
23
+ command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_VIDEO_PATH} --execution-threads 1024"
24
  os.system(command)
25
  solved_time = time.time() - start_time
26
  message = "Successfully solve the GED problem, using time ({:.3f}s).".format(solved_time)
27
 
28
  return message, ROOP_OUTPUT_VIDEO_PATH
29
 
30
+ @spaces.GPU
31
  def handle_roop(
32
  video_path: str,
33
  img_path: str