Update app.py

app.py

@@ -3,44 +3,42 @@ import joblib
 import numpy as np
 import pandas as pd
 
-# Load the model and unique brand values
 model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
-brand_values = unique_values['Brand']
+neighborhood_values = unique_values['Neighborhood']
 
 # Define the prediction function
-def predict(brand, screen_size, resolution_width, resolution_height):
+def predict(neighborhood, house_size, num_rooms):
     # Convert inputs to appropriate types
-    screen_size = float(screen_size)
-    resolution_width = int(resolution_width)
-    resolution_height = int(resolution_height)
+    house_size = float(house_size)
+    num_rooms = int(num_rooms)
     
     # Prepare the input array for prediction
     input_data = pd.DataFrame({
-        'Brand': [brand],
-        'Screen Size': [screen_size],
-        'Resolution (Width)': [resolution_width],
-        'Resolution (Height)': [resolution_height]
+        'Neighborhood': [neighborhood],
+        'House Size': [house_size],
+        'Number of Rooms': [num_rooms]
     })
     
     # Perform the prediction
     prediction = model.predict(input_data)
     
-    return prediction[0]
+    return f"The predicted house price is ${prediction[0]:,.2f}"
 
 # Create the Gradio interface
 interface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.Dropdown(choices=list(brand_values), label="Brand"),
-        gr.Textbox(label="Screen Size"),
-        gr.Textbox(label="Resolution (Width)"),
-        gr.Textbox(label="Resolution (Height)")
+        gr.Dropdown(choices=list(neighborhood_values), label="Neighborhood"),
+        gr.Textbox(label="House Size (in square feet)"),
+        gr.Textbox(label="Number of Rooms")
     ],
     outputs="text",
-    title="Monitor Predictor",
-    description="Enter the brand, screen size, and resolution to predict the target value."
+    title="House Price Predictor",
+    description="Enter the neighborhood, house size, and number of rooms to predict the house price."
 )
 
+
+
 # Launch the app
-interface.launch()
+interface.launch()