Update app.py

app.py

@@ -80,10 +80,10 @@ model.fit(train_ds)
 def process_inputs(education, sex, capital_gains, capital_losses):
   df = pd.DataFrame.from_dict(
       {
-          "education": [edu_in], 
-          "sex": [sex_in],
-          "capital_gains": [cap_gains_in],
-          "capital_losses": [cap_losses_in]    
+          "education": [education], 
+          "sex": [sex],
+          "capital_gains": [capital_gains],
+          "capital_losses": [capital_losses]    
       }
   )
   df = encode_df(df)
@@ -98,12 +98,10 @@ def process_inputs(education, sex, capital_gains, capital_losses):
   pred = model.predict(df)
   if pred > .5:
     pred_bi = 1
-    return {"> $50,000"}
+    return {"> $50,000": pred_bi}
   elif pred <=.5:
     pred_bi = 0
-    return {"<= $50,000"}
-
-
+    return {"<= $50,000": pred_bi}
 
 iface = gr.Interface(
     process_inputs,