Parthebhan
commited on
Commit
•
81c70b4
1
Parent(s):
b7c6db6
Update app.py
Browse files
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
|
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
|