hillol7 commited on
Commit
5b6044b
·
verified ·
1 Parent(s): d37926c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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 ($)", default=50000),
33
- gr.Number(label="Family Size", default=2),
34
- gr.Number(label="Work Experience (years)", default=5)
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