bug fix

app.py

@@ -62,7 +62,7 @@ with open('data/parameters.pkl', 'rb') as f:
 model = TemporalFusionTransformer.load_from_checkpoint('model/tft_check.ckpt', map_location=torch.device('cpu'))
 
 df = pd.read_pickle('data/test_data.pkl')
-df = df.loc[(df["Branch"] == "15") & (df["Group"].isin(["6","7","4","1"]))]
+df = df.loc[(df["Branch"] == 15) & (df["Group"].isin(["6","7","4","1"]))]
 
 rain_mapping = {
     "Yes" : 1,
@@ -101,16 +101,16 @@ datepicker = st.date_input("Start of Forecast", datetime.date(2022, 12, 24), min
 fig, axs = plt.subplots(2, 2, figsize=(8, 6))
 
 # Plot scatter plots for each group
-axs[0, 0].scatter(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'Sales'], color='red', marker='o')
+axs[0, 0].scatter(df.loc[df['Group'] == '4', 'Date'], df.loc[df['Group'] == '4', 'sales'], color='red', marker='o')
 axs[0, 0].set_title('Article Group 1')
 
-axs[0, 1].scatter(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '7', 'Sales'], color='blue', marker='o')
+axs[0, 1].scatter(df.loc[df['Group'] == '7', 'Date'], df.loc[df['Group'] == '7', 'sales'], color='blue', marker='o')
 axs[0, 1].set_title('Article Group 2')
 
-axs[1, 0].scatter(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '1', 'Sales'], color='green', marker='o')
+axs[1, 0].scatter(df.loc[df['Group'] == '1', 'Date'], df.loc[df['Group'] == '1', 'sales'], color='green', marker='o')
 axs[1, 0].set_title('Article Group 3')
 
-axs[1, 1].scatter(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '6', 'Sales'], color='yellow', marker='o')
+axs[1, 1].scatter(df.loc[df['Group'] == '6', 'Date'], df.loc[df['Group'] == '6', 'sales'], color='yellow', marker='o')
 axs[1, 1].set_title('Article Group 4')
 
 # Adjust spacing between subplots