INNER CODE UNIT · Python
read_root
balapriyac/data-science-tutorials · data-science-app/app.py:8
def read_root():
return {"message": "A Simple Prediction API"}
def load_model():
with open('model/classifier.pkl', 'rb') as f:
model = pickle.load(f)
return model
class WineFeatures(BaseModel):
alcohol: float
malic_acid: float
ash: float
alcalinity_of_ash: float
magnesium: float
total_phenols: float
flavanoids: float
nonflavanoid_phenols: float
proanthocyanins: float