fix info write

thearyadev · Feb 16, 2024 · 9f62305 · 9f62305
1 parent 4ba6558
commit 9f62305
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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")