Update app.py

app.py

@@ -999,13 +999,13 @@ with ui.navset_card_tab(id="tab"):
                     multiple=True,
                     selected='compliment'
                 )
-            def plot_loss_rates(df, param_types, loss_types, model_types): 
+            def plot_loss_rates_model(df, param_types, loss_types, model_types): 
                 # interplot each column to be same number of points
                 x = np.linspace(0, 1, 1000)
                 loss_rates = []
                 labels = []
                 #drop the column step
-                df = df.drop(columns=['Step'])
+               # df = df.drop(columns=['Step'])
                 for param_type in param_types:
                     for loss_type in loss_types:
                         for model_type in model_types:
@@ -1028,7 +1028,7 @@ with ui.navset_card_tab(id="tab"):
             fig = None
             df = pd.read_csv('results - denseformer.csv')
             mpl.rcParams.update(mpl.rcParamsDefault)
-            fig = plot_loss_rates(df, input.param_type(),input.loss_type(),input.model_type())
+            fig = plot_loss_rates_model(df, input.param_type(),input.loss_type(),input.model_type())
             return fig
         # @output
         # @render.plot