Parthebhan commited on
Commit
81c70b4
1 Parent(s): b7c6db6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -32,7 +32,7 @@ def salarybracket(age, workclass, education, education_num, marital_status, occu
32
  inputs = np.array([[float(age), float(workclass_value), float(education_value), float(education_num), float(marital_status_value), float(occupation_value), float(relationship_value), float(race_value), float(gender_value), float(capital_gain), float(capital_loss), float(hours_per_week), float(native_country_value)]])
33
  prediction = model.predict(inputs)
34
  prediction_value = prediction[0][0] # Assuming the prediction is a scalar
35
- result = "Income_bracket lesser than or equal to 50K ⬇️" if prediction_value <= 0.5 else "Income_bracket greater than 50K ⬆️"
36
  return f"{result}"
37
 
38
  # Convert mapping to Gradio dropdown options
 
32
  inputs = np.array([[float(age), float(workclass_value), float(education_value), float(education_num), float(marital_status_value), float(occupation_value), float(relationship_value), float(race_value), float(gender_value), float(capital_gain), float(capital_loss), float(hours_per_week), float(native_country_value)]])
33
  prediction = model.predict(inputs)
34
  prediction_value = prediction[0][0] # Assuming the prediction is a scalar
35
+ result = "Income_bracket lesser than or equal to 50K - ↓ " if prediction_value <= 0.5 else "Income_bracket greater than 50K - ↑ "
36
  return f"{result}"
37
 
38
  # Convert mapping to Gradio dropdown options