main.py

import dill as dill
import pandas as pd
from fastapi import FastAPI
from pydantic import BaseModel

app = FastAPI()

with open('model/sberpodpiska.pkl ', 'rb') as file:
    model = dill.load(file)


class Form(BaseModel):
    visit_date: str
    visit_time:  str
    visit_number: int
    utm_source:  str
    utm_medium: str
    utm_campaign:  str
    utm_adcontent:  str
    utm_keyword: str
    device_category:  str
    device_os: str
    device_brand: str
    device_screen_resolution: str
    device_browser:  str
    geo_country: str
    geo_city: str


class Prediction(BaseModel):
    pred: int


@app.get('/status')
def status():
    return "I'm OK"


@app.get('/version')
def version():
    return model['metadata']


@app.post('/predict', response_model=Prediction)
def predict(form: Form):
    df = pd.DataFrame.from_dict([form.dict()])
    y = model['model'].predict(df)

    return {
        'pred': y[0],
    }