Update app.py

app.py

@@ -54,7 +54,7 @@ def predict_score(lat, lon):
     # Unpack into respective values
     mu_pred = mu_pred.numpy().flatten()
 
-    mu_pred2 = xgb_model.predict(inputs.unsqueeze().numpy())
+    mu_pred2 = xgb_model.predict(inputs.unsqueeze(0).numpy())
 
     # r = 1/alpha
     # p = r / (r + mu_pred)