Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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
|
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
|
|
|
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
|