Update app.py

app.py

@@ -17,7 +17,7 @@ def predict(age, workclass, education, marital_status, occupation, relationship,
     # "relationship":[relationship], "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss], 
     # "hours-per-week":[hours_per_week], "native-country":[native_country]}
     columns = {
-    "age": [age], "workclass":[workclass], "educational-num":[education], 
+    "age": [age], "workclass":[workclass], "educational-num":[education], "occupation":[occupation],
     "race":[race], "gender":[gender], "capital-gain":[capital_gain], "capital-loss":[capital_loss], 
     "hours-per-week":[hours_per_week], "native-country":[native_country]}
     df = pd.DataFrame(data=columns)