3

Dockerfile

@@ -15,3 +15,6 @@ COPY . .
 
 # Command to run the application
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]
+
+# Set the environment variable for Matplotlib cache
+ENV MPLCONFIGDIR=/tmp/matplotlib

main.py

@@ -228,8 +228,8 @@ def run_xgboost(df):
 
     # Calculate and print RMSE for the test set
     rmse = np.sqrt(mean_squared_error(y_test, predictions))
-
-    return {"latest_prediction": latest_prediction, "RMSE": rmse}
+    np_float = np.float32(rmse)
+    return {"latest_prediction": latest_prediction, "RMSE": float(np_float)}
 
 
 @app.get("/ticker/{ticker}")