snajmark commited on
Commit
0b2163b
·
1 Parent(s): 2b54252

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -148,7 +148,7 @@ def predict_inverse(hardness_target, ys_target, request: gr.Request):
148
 
149
  continuous_variables = ['PROPERTY: Calculated Density (g/cm$^3$)',
150
  'PROPERTY: Calculated Young modulus (GPa)']
151
- categorical_variables = list(one_hot.columns)
152
  for c in continuous_variables:
153
  categorical_variables.remove(c)
154
 
 
148
 
149
  continuous_variables = ['PROPERTY: Calculated Density (g/cm$^3$)',
150
  'PROPERTY: Calculated Young modulus (GPa)']
151
+ categorical_variables = list(one_hot_columns)
152
  for c in continuous_variables:
153
  categorical_variables.remove(c)
154