Update src/streamlit_app.py

src/streamlit_app.py

@@ -37,4 +37,9 @@ input_data = {
     'Store_Size': store_size,
     'Store_Location_City_Type': store_location_city_type,
     'Store_Type': store_type,
-}
+}
+
+# Predict button
+if st.button("Predict"):
+    result = model.predict(input_data).tolist()
+    st.write(f"Prediction: The customer is likely to **{result}**.")