Ashrafb commited on
Commit
a1a0f9d
·
1 Parent(s): d66018e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -29,18 +29,11 @@ def run_scripts(target, source, mode):
29
  with open(f'temp/source{target_extension}', 'wb') as f:
30
  f.write(source_bytes)
31
 
32
- if mode == "Face Swapper":
33
- cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_swapper"]
34
- subprocess.run(cmd1)
35
-
36
- if mode == "Face Enhancer":
37
- cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
38
- subprocess.run(cmd2)
39
-
40
- if mode == "Both":
41
  cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_swapper"]
42
  subprocess.run(cmd1)
43
 
 
44
  cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
45
  subprocess.run(cmd2)
46
 
 
29
  with open(f'temp/source{target_extension}', 'wb') as f:
30
  f.write(source_bytes)
31
 
32
+ if mode in ["Face Swapper", "Both"]:
 
 
 
 
 
 
 
 
33
  cmd1 = ["python3", "run.py", "-s", f'temp/source{target_extension}', "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_swapper"]
34
  subprocess.run(cmd1)
35
 
36
+ if mode in ["Face Enhancer", "Both"]:
37
  cmd2 = ["python3", "run.py", "-t", f'temp/target{target_extension}', "-o", output_path, "--frame-processor", "face_enhancer"]
38
  subprocess.run(cmd2)
39