Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -22,16 +22,16 @@ def predict_spending_score(annual_income, family_size, work_experience):
|
|
22 |
features = df[['Annual Income ($)', 'Family Size', 'Family_Income_Product', 'Family_Income_Ratio']]
|
23 |
features_scaled = scaler.transform(features)
|
24 |
|
25 |
-
|
26 |
cluster = kmeans.predict(features_scaled)
|
27 |
return f'Cluster: {int(cluster[0])}'
|
28 |
|
29 |
iface = gr.Interface(
|
30 |
fn=predict_spending_score,
|
31 |
inputs=[
|
32 |
-
gr.Number(label="Annual Income ($)",
|
33 |
-
gr.Number(label="Family Size",
|
34 |
-
gr.Number(label="Work Experience (years)",
|
35 |
],
|
36 |
outputs="text",
|
37 |
live=True
|
|
|
22 |
features = df[['Annual Income ($)', 'Family Size', 'Family_Income_Product', 'Family_Income_Ratio']]
|
23 |
features_scaled = scaler.transform(features)
|
24 |
|
25 |
+
|
26 |
cluster = kmeans.predict(features_scaled)
|
27 |
return f'Cluster: {int(cluster[0])}'
|
28 |
|
29 |
iface = gr.Interface(
|
30 |
fn=predict_spending_score,
|
31 |
inputs=[
|
32 |
+
gr.Number(label="Annual Income ($)", value=50000),
|
33 |
+
gr.Number(label="Family Size", value=2),
|
34 |
+
gr.Number(label="Work Experience (years)", value=5)
|
35 |
],
|
36 |
outputs="text",
|
37 |
live=True
|