Abijith commited on
Commit
671b280
1 Parent(s): 79aa5cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -24,7 +24,9 @@ def test_ml_model(longitude, latitude, housing_median_age, total_rooms, total_be
24
  features = {name:np.array(value) for name, value in df_test_norm.items()}
25
 
26
  result = loaded_model.predict(features)
27
- return (f'predicted: {result}')
 
 
28
 
29
  demo = gr.Interface(fn=test_ml_model,
30
  inputs=[gr.Number(value=0.0), gr.Number(value=0.0), gr.Number(value=0.0),
 
24
  features = {name:np.array(value) for name, value in df_test_norm.items()}
25
 
26
  result = loaded_model.predict(features)
27
+ if result['dense'][0][0] > 0.5:
28
+ return 'Houses in this neighborhood above average price'
29
+ return 'Houses in this neighborhood below average price'
30
 
31
  demo = gr.Interface(fn=test_ml_model,
32
  inputs=[gr.Number(value=0.0), gr.Number(value=0.0), gr.Number(value=0.0),