Update app.py

app.py

@@ -60,10 +60,10 @@ def main():
                                                "sex": [sex],
                                                "hours.per.week": [hours_per_week],
                                                "native.country": [native_country]}))
+            result = ">50k" if result[0] == 1 else '<=50k'
+            st.success("Your predicted income is "+result)
 
 if __name__=="__main__":
     main()                                                                            
-            # Show prediction
-            result = ">50k" if result[0] == 1 else '<=50k'
-            st.success("Your predicted income is "+result) 
+
 # Run main()