yongqiang commited on
Commit
2271615
·
1 Parent(s): cbde913

Update: save summary res to .md

Browse files
Files changed (1) hide show
  1. demo.py +9 -1
demo.py CHANGED
@@ -274,4 +274,12 @@ if __name__ == "__main__":
274
  # print("\n\n开始进行最终总结...")
275
  # prompt = "\n".join(output_text_l)
276
  # final_output_text = run_model(prompt)
277
- # print("最终总结结果:\n", final_output_text)
 
 
 
 
 
 
 
 
 
274
  # print("\n\n开始进行最终总结...")
275
  # prompt = "\n".join(output_text_l)
276
  # final_output_text = run_model(prompt)
277
+ # print("最终总结结果:\n", final_output_text)
278
+
279
+ # 将 output_text_l 保存为 md 格式的文件, 按照原始 LLM 的输出格式进行保存
280
+ output_summary_path = os.path.join(args.output_dir, f"{wav_file.split('/')[-1]}_summary.md")
281
+ output_text_combined = "\n\n".join(output_text_l)
282
+ with open(output_summary_path, 'w', encoding='utf-8') as f:
283
+ f.write(output_text_combined)
284
+
285
+ logger.info(f"LLM Inference completed. Summary saved to {output_summary_path}")