jonathanjordan21 commited on
Commit
d705de9
·
verified ·
1 Parent(s): 1040f96

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def predict(lat, lon, column_order):
23
 
24
  input_dict_new = {}
25
 
26
- for col in REQUIRED_COLUMN_ORDER:
27
  input_dict_new[col] = input_dict[col]
28
 
29
  mu_pred = loaded_model.predict([list(input_dict_new.values())])[0]
 
23
 
24
  input_dict_new = {}
25
 
26
+ for col in [x[4:].strip() if x[:4] == 'num_' else x.strip() for x in column_order.replace("'","").replace('"','').split(",")]:
27
  input_dict_new[col] = input_dict[col]
28
 
29
  mu_pred = loaded_model.predict([list(input_dict_new.values())])[0]