Sonny4Sonnix commited on
Commit
bfd5973
·
1 Parent(s): 206327a

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +4 -4
main.py CHANGED
@@ -38,10 +38,10 @@ async def predict(input: InputFeatures):
38
  'Blood_Work_Result_4', 'patients_age']]
39
 
40
  #creating pipelines
41
- num_pipeline= Pipeline([('imputer', SimpleImputer()),('scaler', StandardScaler())])
42
 
43
 
44
- full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
45
 
46
  XGB = Pipeline([
47
  ("col_trans", full_pipeline), # You need to define full_pipeline
@@ -55,8 +55,8 @@ full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
55
 
56
  return prediction
57
 
58
- # if __name__ == '__main__':
59
- # uvicorn.run("Main:app", reload=True)
60
 
61
 
62
 
 
38
  'Blood_Work_Result_4', 'patients_age']]
39
 
40
  #creating pipelines
41
+ num_pipeline= Pipeline([('imputer', SimpleImputer()),('scaler', StandardScaler())])
42
 
43
 
44
+ full_pipeline=ColumnTransformer([('num_pipe',num_pipeline,num_attr)])
45
 
46
  XGB = Pipeline([
47
  ("col_trans", full_pipeline), # You need to define full_pipeline
 
55
 
56
  return prediction
57
 
58
+ if __name__ == '__main__':
59
+ uvicorn.run("Main:app", reload=True)
60
 
61
 
62