Update app.py
Browse files
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(
|
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,
|