Update app.py

app.py

@@ -168,8 +168,8 @@ def prediction():
         scaler = StandardScaler()
         from sklearn.preprocessing import OneHotEncoder
         oneh = OneHotEncoder(handle_unknown="ignore")
-        oneh.fit(xtrain[features])
-        X_test = oneh.transform(report_data[features])
+        oneh.fit(xtrain)
+        X_test = oneh.transform(report_data)
         #xtrain = scaler.fit_transform(xtrain)    
         #report_data=scaler.transform(report_data)
         response = model_xgb.predict(X_test)