yongqiang commited on
Commit ·
2271615
1
Parent(s): cbde913
Update: save summary res to .md
Browse files
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}")
|