Update app.py

app.py

@@ -3,6 +3,7 @@ from fastapi.responses import FileResponse
 from tensorflow.keras.models import load_model, Sequential
 from tensorflow.keras.layers import Dense, LSTM
 from tensorflow.keras.optimizers import Adam
+import traceback
 from sklearn.preprocessing import MinMaxScaler
 import numpy as np
 import tempfile
@@ -30,11 +31,12 @@ async def predict(model: UploadFile = File(...), data: str = None):
         print(data)
         # Process the data
         predictions = model.predict(ds_normalized.reshape(1, 12, 1)).tolist()
-        predictions_rescaled = scaler.inverse_transform(np.array(predictions).reshape(-1, 1)).flatten()  
+        predictions_rescaled = scaler.inverse_transform(predictions).flatten()  
 
-        return {"predictions": predictions_rescaled}
+        return {"predictions": predictions}
 
     except Exception as e:
+        print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=str(e))
 
 @app.post("/retrain")
@@ -74,6 +76,7 @@ async def retrain(model: UploadFile = File(...), data: str = None):
             headers={"Content-Disposition": "attachment; filename=updated_model.h5"}
         )
     except Exception as e:
+        print(traceback.format_exc())
         raise HTTPException(status_code=500, detail=str(e))
     finally:
         # Clean up temporary files