Update app.py

app.py

@@ -5,7 +5,7 @@ import numpy as np
 # Load the model and unique brand values
 model = joblib.load('model.joblib')
 unique_values = joblib.load('unique_values.joblib')
-brand_values = unique_values['Brand']
+brand_values = list(unique_values['Brand'])
 
 # Define the prediction function
 def predict(brand, screen_size, resolution_width, resolution_height):
@@ -26,7 +26,7 @@ def predict(brand, screen_size, resolution_width, resolution_height):
 interface = gr.Interface(
     fn=predict,
     inputs=[
-        gr.Dropdown(choices=["a", "b"], label="Brand"),
+        gr.Dropdown(brand_values, label="Brand"),
         gr.Textbox(label="Screen Size"),
         gr.Textbox(label="Resolution (Width)"),
         gr.Textbox(label="Resolution (Height)")