diff --git a/app.py b/app.py
index 16e0043..0744d4a 100644
--- 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")