修改时间

facefusion/api/core.py

@@ -25,6 +25,7 @@ async def process_frames(params: Params = Body(...)) -> dict:
     # delete_files_in_directory(f"{work_dir}temp/source")
     # delete_files_in_directory(f"{work_dir}temp/target")
     # delete_files_in_directory(f"{work_dir}temp/output")
+    start_time = time.time()
 
     if not (params.source or params.target):
         return {"message": "Source image or path is required"}
@@ -42,7 +43,8 @@ async def process_frames(params: Params = Body(...)) -> dict:
 
     save_file(globals.source_path, params.source)
     save_file(globals.target_path, params.target)
-
+    end_time = time.time()
+    print("程序处理时间：", end_time - start_time)
     try:
         core.api_conditional_process()
     except Exception as e:
@@ -52,9 +54,14 @@ async def process_frames(params: Params = Body(...)) -> dict:
     # delete_files_in_directory(globals.source_path)
     # delete_files_in_directory(globals.target_path)
     # delete_files_in_directory(globals.output_path)
+    end_time2 = time.time()
+    print("生成时间：", end_time2 - end_time)
     os.remove(globals.source_path)
     os.remove(globals.target_path)
     os.remove(globals.output_path)
+    end_time3 = time.time()
+    print("删除文件时间：", end_time3 - end_time2)
+    print("最终运行时间：", end_time3 - start_time)
     return {'output': output}
 
 def update_global_variables(params: Params):