huntrezz commited on
Commit
c874ca6
·
verified ·
1 Parent(s): 82930cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -81,7 +81,10 @@ def predict_total_pay(gender, job_title, ethnicity):
81
 
82
  def gradio_predict(gender, ethnicity, job_title):
83
  predicted_pay = predict_total_pay(gender, job_title, ethnicity)
84
- return f"${predicted_pay:.2f}"
 
 
 
85
 
86
  # Prepare dropdown options
87
  genders = df['GENDER'].dropna().unique().tolist()
 
81
 
82
  def gradio_predict(gender, ethnicity, job_title):
83
  predicted_pay = predict_total_pay(gender, job_title, ethnicity)
84
+ if predicted_pay < 0:
85
+ return "0.00 or less. Discrimination Suspected"
86
+ else:
87
+ return f"${predicted_pay:.2f} per year"
88
 
89
  # Prepare dropdown options
90
  genders = df['GENDER'].dropna().unique().tolist()