ElioBaserga commited on
Commit
65db99b
·
verified ·
1 Parent(s): decdba5

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +27 -0
app.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import pickle
2
+ import pandas as pd
3
+ import gradio as gr
4
+
5
+ eb = "ebas.pkl"
6
+ with open(eb, mode="rb") as f:
7
+ model = pickle.load(f)
8
+
9
+ features = ['rooms', 'area', 'pop', 'pop_dens', 'emp', 'tax_income', 'pop_dens_per_room']
10
+
11
+ def predict(rooms, area, pop, pop_dens, emp, tax_income, pop_dens_per_room):
12
+ input_data = pd.DataFrame([[rooms, area, pop, pop_dens, emp, tax_income, pop_dens_per_room]],
13
+ columns=features)
14
+ prediction = model.predict(input_data)[0]
15
+ return prediction
16
+
17
+ i = gr.Interface(
18
+ fn=predict,
19
+ inputs=["number", "number", "number", "number", "number", "number", "number"],
20
+ outputs="number",
21
+ title="Apartment Rent Price Predictor with Pop_dens_per_room feature",
22
+ description="Enter apartment features to predict the rent price.",
23
+ examples=[[4.5,130,3664,282.9343629344,783,79838,62.874302874311105],
24
+ [3.5,102,8796,605.3682037164,2469,94471,172.9623439189714]]
25
+ )
26
+
27
+ i.launch()