Update app.py

app.py

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