jonathanjordan21 commited on
Commit
831d21f
·
verified ·
1 Parent(s): fa3105f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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():