call inference_gfpgan cmd

colab_flask.py

@@ -122,21 +122,25 @@ def call_gfpgan(wav2lip_mp4, audio_path, output_mp4):
     # -s 2
     # --only_center_face
     # --bg_upsampler None
+    start = datetime.now()
     cmd = [
         "python",
         "gfpgan/inference_gfpgan.py",
         "-i", origin_frames_folder,
         "-o", folder_path,
-        "-v", 1.3,
-        "-s", 2,
+        "-v", str(1.4),
+        "-s", str(2),
         "--only_center_face",
-        "--bg_upsampler", None
+        "--bg_upsampler", 'realesrgan'
     ]
-    call(cmd)
+    print(cmd)
+    proc = subprocess.Popen(cmd, shell=False)
+    proc.communicate()
+    duration = datetime.now() - start
+    print(f'inference_gfpgan finished in {duration}')
 
     restoredFramesPath = os.path.join(folder_path, 'restored_imgs')
     os.makedirs(restoredFramesPath, exist_ok=True)
-    folder_path = folder_path
     dir_list = os.listdir(restoredFramesPath)
     dir_list.sort()
     batch = 0