Update app.py

app.py

@@ -136,7 +136,7 @@ def predict_inverse(hardness_original_target, ys_original_target, request: gr.Re
 
     # Domain 
     domain = []
-    for c in one_hot.columns:
+    for c in one_hot_columns:
         if c in continuous_variables:
             if c.startswith("PROPERTY: Metal") and c not in metals_to_use:
                 domain.append({'name': str(c), 'type': 'continuous', 'domain': (0., 0.)})