Abijith commited on
Commit
71f866a
·
1 Parent(s): a93834e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -20,8 +20,9 @@ def test_ml_model(longitude, latitude, housing_median_age, total_rooms, total_be
20
  total_rooms, total_bedrooms, population,
21
  households, median_income, median_house_value], columns=data_heading)
22
  df_test_norm = pd.DataFrame(scaler(df_test), columns=data_heading)
 
23
 
24
- result = loaded_model.predict(df_test_norm)
25
  return (f'predicted: {result}')
26
 
27
  demo = gr.Interface(fn=test_ml_model,
 
20
  total_rooms, total_bedrooms, population,
21
  households, median_income, median_house_value], columns=data_heading)
22
  df_test_norm = pd.DataFrame(scaler(df_test), columns=data_heading)
23
+ features = {name:np.array(value) for name, value in df_test_norm.items()}
24
 
25
+ result = loaded_model.predict(features)
26
  return (f'predicted: {result}')
27
 
28
  demo = gr.Interface(fn=test_ml_model,