bug fix

app.py

@@ -64,6 +64,7 @@ def generate_plot(df, preds):
     fig, axs = plt.subplots(2, 2, figsize=(8, 6))
     
     df = pd.merge(df, preds, left_on=["time_idx", "Group"], right_on=["pred_idx", "Group"], how = "left")
+    df[["sales", "pred"]] = df[["sales", "pred"]].replace(0.0, np.nan)
     # Plot scatter plots for each group
     axs[0, 0].scatter(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='grey', marker='o')
     axs[0, 0].plot(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'pred'], color = 'red')
@@ -93,7 +94,6 @@ def load_data():
         parameters = pickle.load(f)
     df = pd.read_pickle('data/test_data.pkl')
     df = df.loc[(df["Branch"] == "15") & (df["Group"].isin(["6","7","4","1"]))]
-    df = df.replace(0, np.nan)
     return parameters, df
 
 @st.cache_resource