Update template_gradio_interface.py

template_gradio_interface.py

@@ -58,8 +58,8 @@ def call_predict(inference_dict, cols_order):
         shap.summary_plot(shap_values[0], df_preprocessed, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
         plt.subplot(1,2,2)
         shap.summary_plot(shap_values[1], df_preprocessed, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
-        plt.subplot(1,2,3)
-        shap.summary_plot(shap_values[2], df_preprocessed, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
+        # plt.subplot(1,2,3)
+        # shap.summary_plot(shap_values[2], df_preprocessed, show=False, feature_names=df_preprocessed.columns, plot_size=(15, 15))
         plt.tight_layout()
         plt.subplots_adjust(wspace=2.0)
         fig = plt.gcf()
@@ -67,6 +67,7 @@ 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
+        return y_pred_rescaled[0][0], 10, y_pred_rescaled[0][1], 10, fig
 
     return lambda *x: predict_from_list(x)