heatingma commited on
Commit
1110322
1 Parent(s): 417ab93

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -16,7 +16,8 @@ def _handle_roop_solve(
16
  raise gr.Error("Please upload source video!")
17
  if img_path is None:
18
  raise gr.Error("Please upload target image!")
19
-
 
20
  start_time = time.time()
21
  command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_VIDEO_PATH}"
22
  os.system(command)
 
16
  raise gr.Error("Please upload source video!")
17
  if img_path is None:
18
  raise gr.Error("Please upload target image!")
19
+ if not os.path.exists("media"):
20
+ os.mkdir("media")
21
  start_time = time.time()
22
  command = f"python run.py -t {video_path} -s {img_path} -o {ROOP_OUTPUT_VIDEO_PATH}"
23
  os.system(command)