Update template_gradio_interface.py

template_gradio_interface.py

@@ -57,13 +57,16 @@ def call_predict(inference_dict, cols_order):
     scaler_targets = unpickle_file(inference_dict["inference"]["scaler_targets_path"])
     encoder = unpickle_file(inference_dict["inference"]["encoder_path"])
     explainer = unpickle_file(inference_dict["inference"]["explainer_path"])
+
+    categorical_columns = ["infill_pattern", "material"]
+    target_columns = ["roughness", "tension_strength", "elongation"]
+    numerical_columns = [c for c in cols_order if c not in categorical_columns]
     
 
     def predict_from_list(x_list):
         df = pd.DataFrame([x_list], columns=cols_order)
         print(df.shape)
 
-        df_preprocessed = 
 
         y_pred, _, shap_values = predict(inference_dict["inference"]["model_path"], df_preprocessed, explainer)