bug fix

app.py

@@ -68,7 +68,7 @@ def update_plot(df, preds, axs, fig):
     axs[0, 1].plot(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[1, 0].plot(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
     axs[1, 1].plot(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
-    return st.pyplot(fig)
+    return fig, axs
 
 @st.cache_resource
 def generate_plot(df):