diff --git a/predict.py b/predict.py
index bc62418..de75e9b 100644
--- a/predict.py
+++ b/predict.py
@@ -16,8 +16,5 @@ 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]
 
-with open("prediction.txt", "w") as txt_file:
-    for line in pre_array:
-        txt_file.write(f"{line}")
\ No newline at end of file
+pre.tofile('prediction.csv', sep=',', format='%s')
\ No newline at end of file