Update template_gradio_interface.py

template_gradio_interface.py

@@ -30,7 +30,8 @@ def call_predict(inference_dict, cols_order):
     # explainer = unpickle_file(inference_dict["explainer_path"])
 
     categorical_columns = ["infill_pattern", "material"]
-    target_columns = ["roughness", "tension_strength", "elongation"]
+    # target_columns = ["roughness", "tension_strength", "elongation"]
+    target_columns = ["roughness", "tension_strength"]
     numerical_columns = [c for c in cols_order if c not in categorical_columns]
 
     df_train = pd.read_csv("dataset_preprocessed.csv", sep=";")