ElioBaserga's picture
Upload app.py
65db99b verified
raw
history blame contribute delete
985 Bytes
import pickle
import pandas as pd
import gradio as gr
eb = "ebas.pkl"
with open(eb, mode="rb") as f:
model = pickle.load(f)
features = ['rooms', 'area', 'pop', 'pop_dens', 'emp', 'tax_income', 'pop_dens_per_room']
def predict(rooms, area, pop, pop_dens, emp, tax_income, pop_dens_per_room):
input_data = pd.DataFrame([[rooms, area, pop, pop_dens, emp, tax_income, pop_dens_per_room]],
columns=features)
prediction = model.predict(input_data)[0]
return prediction
i = gr.Interface(
fn=predict,
inputs=["number", "number", "number", "number", "number", "number", "number"],
outputs="number",
title="Apartment Rent Price Predictor with Pop_dens_per_room feature",
description="Enter apartment features to predict the rent price.",
examples=[[4.5,130,3664,282.9343629344,783,79838,62.874302874311105],
[3.5,102,8796,605.3682037164,2469,94471,172.9623439189714]]
)
i.launch()