Update train_lstm_model.py

train_lstm_model.py

@@ -6,6 +6,10 @@ from tensorflow.keras.layers import LSTM, Dense
 from tensorflow.keras.optimizers import Adam
 import os
 import shutil
+import tensorflow as tf
+
+# Disable GPU to avoid cuDNN/cuBLAS conflicts (for now)
+os.environ["CUDA_VISIBLE_DEVICES"] = ""  # Force TensorFlow to use CPU
 
 # Check disk space
 available_space = shutil.disk_usage('.').free / (1024 * 1024)  # Free space in MB
@@ -59,8 +63,13 @@ os.makedirs('models', exist_ok=True)
 if not os.access('models', os.W_OK):
     raise PermissionError("No write permissions for the 'models' directory.")
 
-# Save the trained model
+# Remove any existing (invalid) model file
 model_path = 'models/lstm_energy_model.h5'
+if os.path.exists(model_path):
+    os.remove(model_path)
+    print(f"Removed existing (invalid) file at {model_path}")
+
+# Save the trained model
 try:
     model.save(model_path)
 except Exception as e: