import logging
import pickle

# Server
import uvicorn
from fastapi import FastAPI

app = FastAPI()

my_logger = logging.getLogger()
logging.basicConfig(
    level=logging.INFO,
    filename='model-service.log',
    filemode='a',
    )

# Initialize files
clf = pickle.load(open('{{ filename }}', 'rb'))

@app.post("/predict")
def predict(data: dict):

    X = [list(data.values())]

    prediction = clf.predict(X)

    return {"prediction": float(prediction[0])}