Spaces:
Runtime error
Runtime error
import gradio as gr | |
import numpy as np | |
import pandas as pd | |
from sklearn.linear_model import LinearRegression | |
import pickle | |
with open('model.pkl', 'rb') as file: | |
model = pickle.load(file) | |
def proLocation(location): | |
if location=='Rural': | |
return 0 | |
elif location=='Urban': | |
return 1 | |
else: | |
return 2 | |
def predict_loan_amount(gender, age, income, income_stability, property_age, property_price, property_location): | |
input_data = { | |
"Gender": [1 if gender == 'M' else 0], | |
"Age": [age], | |
"Income (USD)": [income], | |
"Income Stability": [1 if income_stability == 'Low' else 0], | |
"Property Age": [property_age], | |
"Property Price": [property_price], | |
"Property Location": [proLocation(property_location)], | |
} | |
input_df = pd.DataFrame(input_data) | |
prediction = model.predict(input_df.to_numpy()) | |
return prediction[0] | |
# Gradio interface | |
iface = gr.Interface( | |
fn=predict_loan_amount, | |
inputs=[ | |
gr.Radio(['F', 'M'], label='Gender'), | |
gr.Slider(18, 70, step=1, label='Age'), | |
gr.Number(label='Income (USD)'), | |
gr.Radio(['Low', 'High'], label='Income Stability'), | |
gr.Number(label='Property Age'), | |
gr.Number(label='Property Price'), | |
gr.Radio(['Rural', 'Urban', 'Semi-Urban'], label='Property Location'), | |
], | |
outputs="number", | |
live=True | |
) | |
iface.launch() | |