Update app.py

app.py

@@ -29,7 +29,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)
+    num_banks = inputs.pop("num_banks_in_radius", 0)
 
     inputs = torch.tensor([lat,lon] + list(inputs.values()))