Update app.py

app.py

@@ -53,26 +53,27 @@ def main():
         hours_per_week = st.slider("Hours per week", min_value=1, max_value=100)
         native_country = st.selectbox("Country",options =  unique_country)
 
-     clicked = st.form_submit_button("Predict income")
-        if clicked:
-            education_encoded = EDU_DICT.get(education, 0)  # Default to 0 if education is not found
-            result = model.predict(pd.DataFrame({
-                "age": [age],
-                "workclass": [workclass],
-                "fnlwgt": [fnlwgt],
-                "education": [education_encoded],
-                "educational-num": [educational_num],
-                "marital.status": [marital_status],
-                "occupation": [occupation],
-                "relationship": [relationship],
-                "race": [race],
-                "gender": [gender],
-                "capital_gain": [capital_gain],
-                "capital_loss": [capital_loss],
-                "hours.per.week": [hours_per_week],
-                "native.country": [native_country]}))
-            result = '>50K' if result[0] == 1 else '<=50K'
-              st.success('The predicted income is {}'.format(result))
+     if clicked:
+        education_encoded = EDU_DICT.get(education, 0)  # Default to 0 if education is not found
+        result = model.predict(pd.DataFrame({
+            "age": [age],
+            "workclass": [workclass],
+            "fnlwgt": [fnlwgt],
+            "education": [education_encoded],
+            "educational-num": [educational_num],
+            "marital.status": [marital_status],
+            "occupation": [occupation],
+            "relationship": [relationship],
+            "race": [race],
+            "gender": [gender],
+            "capital_gain": [capital_gain],
+            "capital_loss": [capital_loss],
+            "hours.per.week": [hours_per_week],
+            "native.country": [native_country]
+        }))
+    result = '>50K' if result[0] == 1 else '<=50K'
+    st.success('The predicted income is {}'.format(result))
+
 
 if __name__ == '__main__':