Update app.py

app.py

@@ -40,7 +40,7 @@ with tab1:
     col1, col2 = st.columns(2)
 
     with col1:
-        st.markdown("Parking place near building:")
+        st.markdown("**Parking place near building:**")
 
         # Function to load the building model
         
@@ -74,7 +74,7 @@ with tab1:
         st.dataframe(building_new[['prediction']].tail(5))
 
     with col2:
-        st.markdown("Parking place near bikelane:")
+        st.markdown("**Parking place near bikelane:**")
             
         # Function to load the bikelane model
         @st.cache_data()
@@ -124,21 +124,34 @@ with tab1:
         st.experimental_rerun()
 
 with tab2:
-    st.markdown('Experiment with building model')
-    x_input_building = st.slider("Choose your x-value", -232, 909, 0)
-    y_input_building = st.slider("Choose your y-value", -1112, 435, 0)
-    z_input_building = st.slider("Choose your z-value", -1648, 226, 0)
-
     def explore_magnetic_field(model, x, y, z):
         input_list = [x, y, z]
         res = model.predict(np.asarray(input_list).reshape(1,-1))
         explorer_prediction = res[0]
-        return explorer_prediction
+        if explorer_prediction == 'detection':
+            label = "Vehicle detected"
+        else:
+            label = "No vehicle detected"
+        return label
+        
+    st.markdown('**Experiment with building model:**')
+    x_input_building = st.slider("Choose your x-value", -232, 909, 0)
+    y_input_building = st.slider("Choose your y-value", -1112, 435, 0)
+    z_input_building = st.slider("Choose your z-value", -1648, 226, 0)
 
     if st.button("Predict"):
         building_input_prediction = explore_magnetic_field(building_hist_model, x_input_building, y_input_building, z_input_building)
         st.write(building_input_prediction)
 
+    st.markdown('**Experiment with bikelane model:**')
+    x_input_bikelane = st.slider("Choose your x-value", -547, 288, 0)
+    y_input_bikelane = st.slider("Choose your y-value", -1007, 786, 0)
+    z_input_bikelane = st.slider("Choose your z-value", -1475, 16, 0)
+
+    if st.button("Predict"):
+        bikelane_input_prediction = explore_magnetic_field(bikelane_hist_model, x_input_bikelane, y_input_bikelane, z_input_bikelane)
+        st.write(bikelane_input_prediction)
+
 with tab3:
     st.markdown('This application is made as part of the module "Data Engineering and Machine Learning Operations in Business - F2024" in Business Data Science 2nd Semester at Aalborg University Business School.')
     st.markdown('The application is made by Annika and Mikkel and is divided into 4 tabs:')