diff --git a/predict.py b/predict.py
index 4b36927..bc62418 100644
--- a/predict.py
+++ b/predict.py
@@ -18,4 +18,6 @@ accu = reloaded.evaluate(x_to_test, y_to_test)
 pre = reloaded.predict(x_to_test)
 pre_array = [float(i)/sum(pre) for i in pre]
 
-pre_array.tofile('prediction.csv', sep=',', format='%s')
\ No newline at end of file
+with open("prediction.txt", "w") as txt_file:
+    for line in pre_array:
+        txt_file.write(f"{line}")
\ No newline at end of file