Update template_gradio_interface.py

template_gradio_interface.py

@@ -30,8 +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"]
+    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=";")
@@ -67,7 +67,9 @@ def call_predict(inference_dict, cols_order):
 
         print("mmmmmmmmmmmmmmmmmmmmm")
         print(y_pred_rescaled.shape)
-        return y_pred_rescaled[0][0], 10, y_pred_rescaled[0][1], 10, y_pred_rescaled[0][2], 10, fig
+        print(y_pred_rescaled[0][0])
+        print(y_pred_rescaled[0][1])
+        # return y_pred_rescaled[0][0], 10, y_pred_rescaled[0][1], 10, y_pred_rescaled[0][2], 10, fig
         return y_pred_rescaled[0][0], 10, y_pred_rescaled[0][1], 10, fig
 
     return lambda *x: predict_from_list(x)