Update app.py

app.py

@@ -38,6 +38,7 @@ def predict_score(lat, lon):
     # Convert input to tensor
     # inputs = torch.tensor([[lat, lon]], dtype=torch.float32)
     inputs = compute_features((lat,lon))
+    num_banks = inputs.get("num_banks_in_radius", 0)
 
     inputs = torch.tensor([lat,lon] + list(inputs.values()))
 
@@ -53,7 +54,8 @@ def predict_score(lat, lon):
     # You can apply any post-processing here
     return {
         "Score": round(float(score), 3),
-        "Num Banks": round(float(mu_pred), 3),
+        "Number of current ATMs": round(float(mu_pred), 3),
+        "Number of ideal ATMs" : num_banks_in_radius
         # "Normal Score": round(float(normal_score), 3),
     }
 
@@ -66,7 +68,8 @@ interface = gr.Interface(
     ],
     outputs=[
         gr.Number(label="Score"),
-        gr.Number(label="Num Banks"),
+        gr.Number(label="Num Current Banks"),
+        gr.Number(label="Num Ideal Banks")
         # gr.Number(label="Normal Score"),
     ],
     title="Bank Location Scoring Model",