zohaibterminator commited on
Commit
9c7aa7b
·
verified ·
1 Parent(s): cbe809b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -4,6 +4,7 @@ import numpy as np
4
  from pydantic import BaseModel
5
  from typing import List
6
  from fastapi import HTTPException
 
7
 
8
  def load_preprocessing_components():
9
  with open("encoder.pkl", "rb") as f:
@@ -12,6 +13,8 @@ def load_preprocessing_components():
12
  scaler = dill.load(f)
13
  return encoder, scaler
14
 
 
 
15
 
16
  class InferenceData(BaseModel):
17
  Age: float
 
4
  from pydantic import BaseModel
5
  from typing import List
6
  from fastapi import HTTPException
7
+ from model_load_save import load_model
8
 
9
  def load_preprocessing_components():
10
  with open("encoder.pkl", "rb") as f:
 
13
  scaler = dill.load(f)
14
  return encoder, scaler
15
 
16
+ model = load_model()
17
+ encoder, scaler = load_preprocessing_components()
18
 
19
  class InferenceData(BaseModel):
20
  Age: float