Update main.py

main.py

@@ -143,9 +143,9 @@ def read_item(ticker: str):
     df = data_manipolation(df,ticker) 
     df=df.round(2)
     Xtrain, ytrain, Xtest, ytest = data_split_train_test(df,ticker)
-    forecast_value = xgb_training_forecast(Xtrain, ytrain, Xtest, ytest).round(2)
+    forecast_value = xgb_training_forecast(Xtrain, ytrain, Xtest, ytest)
     #result = forecast_value.to_json(orient="records")
-    return json.dumps(forecast_value, cls=NumpyEncoder)
+    return json.dumps(forecast_value.round(2), cls=NumpyEncoder)
 
 class NumpyEncoder(json.JSONEncoder):
     """Custom encoder for numpy data types"""