A-New-Day-001 commited on
Commit
9e453e5
1 Parent(s): 8f5ea1d

Update screens/predict.py

Browse files
Files changed (1) hide show
  1. screens/predict.py +2 -2
screens/predict.py CHANGED
@@ -11,13 +11,13 @@ def predict_page():
11
  st.title("💵 Property Price Estimator")
12
 
13
  # User role selection (buyer or seller)
14
- user_role = st.selectbox("Select Your Role", options=["Buyer", "Seller"])
15
 
16
  # Load geocoder and model based on user role
17
  @st.cache(allow_output_mutation=True)
18
 
19
 
20
- if user_role == "Buyer":
21
  model_path = "models/mm-nlp-image-transformer/"
22
  else:
23
  model_path = "models/tabular/"
 
11
  st.title("💵 Property Price Estimator")
12
 
13
  # User role selection (buyer or seller)
14
+ user_role = st.selectbox("Select Your Role", options=("Buyer", "Seller"), index=0)
15
 
16
  # Load geocoder and model based on user role
17
  @st.cache(allow_output_mutation=True)
18
 
19
 
20
+ if user_role == "Seller":
21
  model_path = "models/mm-nlp-image-transformer/"
22
  else:
23
  model_path = "models/tabular/"