Update ReadME

README.md

@@ -84,4 +84,25 @@ def generate_predictions(model, tokenizer, input_file, output_file):
         for result in results:
             json.dump(result, f, ensure_ascii=False)
             f.write('\n')
+
+def main():
+    # GPUメモリのクリア
+    torch.cuda.empty_cache()
+
+    # 時間計測の追加
+    import time
+    start_time = time.time()
+
+    model, tokenizer = load_model()
+    input_file = "{$file_path}"
+    output_file = os.path.join(BASE_DIR, "{$file_path}")
+
+    generate_predictions(model, tokenizer, input_file, output_file)
+
+    # 実行時間の表示
+    elapsed_time = time.time() - start_time
+    print(f"総実行時間: {elapsed_time / 60:.2f}分")
+
+if __name__ == "__main__":
+    main()
 ```