Update app.py (#3)

- Update app.py (129d049b83d88cb2cc84b81f3254faeea05d92d1)


Co-authored-by: Aaleia <Aaleia@users.noreply.huggingface.co>

app.py

@@ -85,18 +85,17 @@ if 'selected' not in st.session_state or st.session_state.selected != selected:
 st.header("Custom Prediction")
 st.write("Enter your own values for prediction:")
 
-selected_features = ["INITIAL_CALL_TYPE", "INITIAL_SEVERITY_LEVEL_CODE", "INCIDENT_DATETIME", "ZIPCODE", "POLICEPRECINCT"]
+selected_features = ["INITIAL_CALL_TYPE", "INITIAL_SEVERITY_LEVEL_CODE", "INCIDENT_DATETIME", "INCIDENT_CLOSE_DATETIME", "ZIPCODE", "POLICEPRECINCT", "FINAL_CALL_TYPE"]
 feature_inputs = {}
 for feature in selected_features:
-    if feature not in ["FINAL_CALL_TYPE"]:
-        
+    if feature not in ["FINAL_CALL_TYPE"]: # ensure final not included
         if feature in ['INITIAL_CALL_TYPE', 'DAY_OF_WEEK', 'POLICEPRECINCT', 'ZIPCODE']:  # categorical
             options = list(sample_data[feature].unique())
         else: # numerical
             options = [str(x) for x in sample_data[feature].unique()]
         feature_inputs[feature] = st.selectbox(feature, options=options)
 
-# Encode user input
+# encode user input
 user_data = encode(pd.DataFrame(feature_inputs, index=[0]))
 
 if st.button("Predict"):