From 9f62305871e3c4d0417baebe6c5f1231519f65eb Mon Sep 17 00:00:00 2001
From: Aryan Kothari <87589047+thearyadev@users.noreply.github.com>
Date: Thu, 15 Feb 2024 19:01:34 -0500
Subject: [PATCH] fix info write

---
 classifier/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/classifier/train.py b/classifier/train.py
index e900652d..97f08cf3 100644
--- a/classifier/train.py
+++ b/classifier/train.py
@@ -104,7 +104,7 @@ def main() -> int:
     shutil.copy(Path("./classifier/model.py"), model_dir / "frozen_model.py")
 
     with open(model_dir / "detail", "w+") as file:
-        file.writelines(info_file)
+        file.write("\n".join(info_file))
 
     with open(model_dir / "__init__.py", "w+") as file:
         file.write("from .frozen_model import transformer, NNModel as FrozenNeuralNetworkModel")