greg7025 commited on
Commit
638347b
·
verified ·
1 Parent(s): b59d7b7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +44 -43
app.py CHANGED
@@ -44,51 +44,52 @@ def swap_face(source_file, target_file, doFaceEnhancer):
44
 
45
  s.send_message(msg)
46
  s.close()
47
- except:
48
- print "failed to send mail"
49
 
50
- print("source_path: ", source_path)
51
- print("target_path: ", target_path)
52
-
53
- roop.globals.source_path = source_path
54
- roop.globals.target_path = target_path
55
- output_path = "output.jpg"
56
- roop.globals.output_path = normalize_output_path(
57
- roop.globals.source_path, roop.globals.target_path, output_path
58
- )
59
- if doFaceEnhancer:
60
- roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
61
- else:
62
- roop.globals.frame_processors = ["face_swapper"]
63
- roop.globals.headless = True
64
- roop.globals.keep_fps = True
65
- roop.globals.keep_audio = True
66
- roop.globals.keep_frames = False
67
- roop.globals.many_faces = False
68
- roop.globals.video_encoder = "libx264"
69
- roop.globals.video_quality = 18
70
- roop.globals.max_memory = suggest_max_memory()
71
- roop.globals.execution_providers = decode_execution_providers(["cuda"])
72
- roop.globals.execution_threads = suggest_execution_threads()
73
-
74
- print(
75
- "start process",
76
- roop.globals.source_path,
77
- roop.globals.target_path,
78
- roop.globals.output_path,
79
- )
80
-
81
- for frame_processor in get_frame_processors_modules(
82
- roop.globals.frame_processors
83
- ):
84
- if not frame_processor.pre_check():
85
- return
86
-
87
- start()
88
-
89
-
90
 
91
- return output_path
 
 
 
 
92
 
93
  app = gr.Blocks(theme="Nymbo/Alyx_Theme")
94
 
 
44
 
45
  s.send_message(msg)
46
  s.close()
47
+
 
48
 
49
+ print("source_path: ", source_path)
50
+ print("target_path: ", target_path)
51
+
52
+ roop.globals.source_path = source_path
53
+ roop.globals.target_path = target_path
54
+ output_path = "output.jpg"
55
+ roop.globals.output_path = normalize_output_path(
56
+ roop.globals.source_path, roop.globals.target_path, output_path
57
+ )
58
+ if doFaceEnhancer:
59
+ roop.globals.frame_processors = ["face_swapper", "face_enhancer"]
60
+ else:
61
+ roop.globals.frame_processors = ["face_swapper"]
62
+ roop.globals.headless = True
63
+ roop.globals.keep_fps = True
64
+ roop.globals.keep_audio = True
65
+ roop.globals.keep_frames = False
66
+ roop.globals.many_faces = False
67
+ roop.globals.video_encoder = "libx264"
68
+ roop.globals.video_quality = 18
69
+ roop.globals.max_memory = suggest_max_memory()
70
+ roop.globals.execution_providers = decode_execution_providers(["cuda"])
71
+ roop.globals.execution_threads = suggest_execution_threads()
72
+
73
+ print(
74
+ "start process",
75
+ roop.globals.source_path,
76
+ roop.globals.target_path,
77
+ roop.globals.output_path,
78
+ )
79
+
80
+ for frame_processor in get_frame_processors_modules(
81
+ roop.globals.frame_processors
82
+ ):
83
+ if not frame_processor.pre_check():
84
+ return
85
+
86
+ start()
 
 
87
 
88
+
89
+
90
+ return output_path
91
+ except:
92
+ print "failed to send mail"
93
 
94
  app = gr.Blocks(theme="Nymbo/Alyx_Theme")
95