Update app.py

app.py

@@ -97,9 +97,11 @@ def process_inputs(education, sex, capital_gains, capital_losses):
   
   pred = model.predict(df)
   if pred > .5:
-    return {"> $50,000": pred}
-  elif pred <=.5:
-    return {"<= $50,000": pred}
+    pred_bi = 1
+    return {"> $50,000": pred_bi}
+  else:
+    pred_bi = 0
+    return {"<= $50,000": pred_bi}
 
 iface = gr.Interface(
     process_inputs,