try around plot

asapdiscovery · Oct 9, 2024 · 8e1839a · 8e1839a
1 parent e255bec
commit 8e1839a
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/app.py b/app.py
@@ -206,8 +206,11 @@ def convert_df(df):
     err = predictions[:, 1]  # rejoin with the original dataframe
 
 # check if the model has a plot prepared
-plot_url = model.pull_plot("test_performance.png", return_as="url")
-
+try:
+    plot_url = model.pull_plot("test_performance.png", return_as="url")
+except:
+    plot_url = None
+
 if plot_url:
     # we are showing the latest test set performance plot
     st.markdown("### Model test set performance")