edit app.py

app.py

@@ -18,9 +18,9 @@ features = ['Gender','Customer Type','Age','Type of Travel','Class','Flight Dist
                 'Cleanliness','Departure Delay in Minutes','Arrival Delay in Minutes']
 
 def infer(data_infer):
-    # predict result random forest model
-    y_pred_rf = catb_pipe.predict(data_infer)
-    return y_pred_rf 
+    # predict result with pre-trained model
+    pred = catb_pipe.predict(data_infer)
+    return pred
 
 # header deployment
 st.header("Predicting Passenger Flight Satisfaction")