TM9450 commited on
Commit
8272fc0
·
1 Parent(s): 1a99606

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -61,6 +61,8 @@ def main():
61
  "hours.per.week": [hours_per_week],
62
  "native.country": [native_country]}))
63
  # Show prediction
 
 
64
 
65
  # Run main()
66
  if __name__ == "__main__":
 
61
  "hours.per.week": [hours_per_week],
62
  "native.country": [native_country]}))
63
  # Show prediction
64
+ result = '>50K' if result[0] == 1 else '<=50K'
65
+ st.success("Your predicted income is "+result)
66
 
67
  # Run main()
68
  if __name__ == "__main__":