Update app.py

app.py

@@ -4,6 +4,12 @@ from fastapi import FastAPI, HTTPException
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, Field
 from model_pipeline import Predictor, FEATURE_MAP
+import io
+import numpy as np
+import matplotlib
+matplotlib.use("Agg")
+import matplotlib.pyplot as plt
+from fastapi.responses import Response
 
 logging.basicConfig(
     level=logging.INFO,