albin commited on
Commit
f5e9881
·
1 Parent(s): 0f41c4f

correct bugs

Browse files
Files changed (1) hide show
  1. main.py +2 -3
main.py CHANGED
@@ -85,9 +85,8 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
85
 
86
  # Apply features encoding (convert everything into int64)
87
  for column in selected_columns:
88
- if column in label_encoders:
89
- encoder = label_encoders[column]
90
- dataFrame_features[column] = encoder.transform(dataFrame_features[column])
91
 
92
  data = dataFrame_features[selected_columns].values
93
 
 
85
 
86
  # Apply features encoding (convert everything into int64)
87
  for column in selected_columns:
88
+ encoder = label_encoders[column]
89
+ dataFrame_features[column] = encoder.transform(dataFrame_features[column])
 
90
 
91
  data = dataFrame_features[selected_columns].values
92