Upload app.py

app.py

@@ -2,14 +2,20 @@ import joblib
 import pandas as pd
 import streamlit as st
 
-# โหลดโมเดลและข้อมูลที่จำเป็น
+Pros= {'Engineer': 1,
+               'Healthcare': 2,
+               'Executive': 3,
+               'Doctor': 4,
+               'Artist': 5,
+               'Lawyer': 6,
+               'Entertainment': 7,
+               'Homemaker': 8,
+               'Marketing': 9}
 model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
 
 def main():
     st.title("Customer Segmentation Prediction")
-
-    # สร้างฟอร์มสำหรับป้อนข้อมูล
     with st.form("questionnaire"):
         Gender = st.selectbox("Gender", unique_values["Gender"])
         Ever_Married = st.selectbox("Ever Married", unique_values["Ever_Married"])
@@ -22,10 +28,8 @@ def main():
         Var_1 = st.selectbox("Var_1", unique_values["Var_1"])
         ID = st.slider("ID", min_value=458982, max_value=467974)
 
-        # สร้างปุ่มสำหรับการทำนาย
         clicked = st.form_submit_button("Predict Segmentation")
         if clicked:
-            # ใช้โมเดลทำนาย Segmentation จากข้อมูลที่ป้อน
             result = model.predict(pd.DataFrame({"Gender": [Gender],
                                                  "Ever_Married": [Ever_Married],
                                                  "Age": [Age],
@@ -37,7 +41,6 @@ def main():
                                                  "Family_Size": [Family_Size],
                                                  "Var_1": [Var_1]
                                                  }))
-            # แปลงผลลัพธ์ให้เป็นข้อความ
             if result[0] == 0:
                 result = "A"
             elif result[0] == 1:
@@ -46,7 +49,6 @@ def main():
                 result = "C"
             else:
                 result = "D"
-            # แสดงผลลัพธ์
             st.success('Predicted Segmentation: {}'.format(result))
 
 if __name__ == '__main__':