Update app.py

app.py

@@ -29,9 +29,9 @@ 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(predictions.reshape(-1, 1)).flatten()  
+        #predictions_rescaled = scaler.inverse_transform(predictions.reshape(-1, 1)).flatten()  
 
-        return {"predictions": predictions_rescaled}
+        return {"predictions": predictions}
 
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
@@ -53,7 +53,7 @@ async def retrain(model: UploadFile = File(...), data: str = None):
         ds = np.array(ds).reshape(-1, 1)
         # Normalize the data
         scaler = MinMaxScaler()
-        dataset_normalized = scaler.fit_transform(ds)
+        ds_normalized = scaler.fit_transform(ds)
 
         x_train = np.array([ds_normalized[i - 12:i] for i in range(12, len(ds_normalized))])
         y_train = ds_normalized[12:]