Merge branch 'main' of https://huggingface.co/spaces/greencatted/EMS-Call-Classifier

app.py

@@ -80,6 +80,34 @@ if 'selected' not in st.session_state or st.session_state.selected != selected:
 	st.session_state.selected = selected
 
 
+# CUSTOM PREDICTIONS SECTION
+
+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"]
+feature_inputs = {}
+for feature in selected_features:
+    if feature not in ["FINAL_CALL_TYPE"]:
+        
+        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
+user_data = encode(pd.DataFrame(feature_inputs, index=[0]))
+
+if st.button("Predict"):
+    t1 = time.time() * 1000
+    predicted_label = predict(user_data)
+    t2 = time.time() * 1000
+
+    st.write("Predicted Call Type:", predicted_label)
+    st.write(f"Prediction Time: {t2-t1:.2f}ms")
+
+
 st.header("Results")
 st.markdown(
 """