Update template_gradio_interface.py

template_gradio_interface.py

@@ -54,8 +54,10 @@ def call_predict(inference_dict, cols_order):
         plt.clf()
         shap.summary_plot(shap_values[0], feature_names=df_preprocessed.columns)
         fig = plt.gcf()
-        
-        return y_pred[0], 10, y_pred[1], 10, y_pred[2], 10, fig
+
+        print("mmmmmmmmmmmmmmmmmmmmm")
+        print(y_pred.shape)
+        return y_pred[0][0], 10, y_pred[0][1], 10, y_pred[0][2], 10, fig
 
     return lambda *x: predict_from_list(x)