Update main.py

main.py

@@ -1,10 +1,24 @@
 # %%
 import sys 
 
+class Tee:
+    def __init__(self, file_name, mode='w'):
+        self.file = open(file_name, mode)
+        self.stdout = sys.stdout
+
+    def write(self, message):
+        self.file.write(message)
+        self.stdout.write(message)
+
+    def flush(self):
+        self.file.flush()
+        self.stdout.flush()
+        
 f= open("log.txt",'a')
+ttt = Tee(f)
 
-sys.stdout = f
-sys.stderr = f
+sys.stdout =ttt
+sys.stderr =ttt
 
 import os 
 os.system("python3 -m http.server 7860 -b 0.0.0.0 &")
@@ -786,8 +800,11 @@ k = model.fit(X_train,
 
 
 model.save("spam-model.h5",include_optimizer=True)
-while(1):
-    pass
+
+import time 
+print("MODELLLLLLL save ")
+os.system('cp log.txt old.txt')
+time.wait(99999999)