Update app.py

app.py

@@ -2,6 +2,11 @@ import gradio as gr
 import pandas as pd 
 import numpy as np 
 
+
+import pickle
+with open("linear_regression_model.pkl", "rb") as file:
+    lin_reg = pickle.load(file)
+    
 # define the input module 
 inputs = [
     gr.inputs.Number(label="Longitude"),
@@ -37,11 +42,8 @@ def predict_price(longitude, latitude, housing_median_age, total_rooms, total_be
 
     features=scaler.transform(features)
     linear_regression_prediction = lin_reg.predict(features)[0]
-    decision_tree_prediction = tree_reg.predict(features)[0]
+    #decision_tree_prediction = tree_reg.predict(features)[0]
 
-    return {
-        'Linear Regression Prediction': linear_regression_prediction,
-        'Decision Tree Prediction': decision_tree_prediction
-    }
+    return linear_regression_prediction
 
     interface = gr.Interface(fn=predict_price, inputs=inputs, outputs=outputs).launch()