Update app.py

app.py

@@ -1054,10 +1054,10 @@ with ui.navset_card_tab(id="tab"):
                     "Select Loss Type:",
                     ["compliment", "cross_entropy", "headless", "2d", "2d_representation_MSEPlusCE"],
                     multiple=False,
-                    selected="cross_entropy"
+                    selected=["cross_entropy"]
                 )
         def plot_loss_rates_model_scale(df, loss_type, model_types):
-            df = df[df['loss_type'] == loss_type]
+            df = df[df['loss_type'] == loss_type[0]]
             # interplot each column to be same number of points
             params = []
             loss_rates = []
@@ -1092,7 +1092,7 @@ with ui.navset_card_tab(id="tab"):
             fig = None
             df = pd.read_csv('training_data_5.csv')
             mpl.rcParams.update(mpl.rcParamsDefault)
-            fig = plot_loss_rates_model_scale(df= df,loss_type= input.loss_type_scale()[0],model_types= input.model_type_scale())
+            fig = plot_loss_rates_model_scale(df= df,loss_type= input.loss_type_scale(),model_types= input.model_type_scale())
             return fig
         # @output
         # @render.plot