Update app.py
Browse files
app.py
CHANGED
|
@@ -31,7 +31,7 @@ def predict_score(lat, lon):
|
|
| 31 |
inputs = compute_features((lat,lon))
|
| 32 |
num_banks = inputs.pop("num_banks_in_radius", 0)
|
| 33 |
|
| 34 |
-
inputs = torch.tensor([lat,lon] + list(inputs.values()))
|
| 35 |
|
| 36 |
# Get model output
|
| 37 |
with torch.no_grad():
|
|
|
|
| 31 |
inputs = compute_features((lat,lon))
|
| 32 |
num_banks = inputs.pop("num_banks_in_radius", 0)
|
| 33 |
|
| 34 |
+
inputs = torch.tensor([lat,lon] + list(inputs.values()), dtype=torch.float32)
|
| 35 |
|
| 36 |
# Get model output
|
| 37 |
with torch.no_grad():
|