Update app.py

app.py

@@ -1028,11 +1028,14 @@ with ui.navset_card_tab(id="tab"):
             for i, loss_rate in enumerate(loss_rates):
                 df_madmad = pd.DataFrame({'x':x, 'loss':loss_rate})
         
-                df_madmad = df_madmad.sort_values(by='x')
+                # df_madmad = df_madmad.sort_values(by='x')
                 df_madmad = df_madmad[df_madmad['x']>x_filter]
                 x = df_madmad['x'].to_list()
                 loss_rate = df_madmad['loss'].to_list()
-                ax.plot(x, loss_rate, label=labels[i])
+                try: 
+                    ax.plot(x, loss_rate, label=labels[i])
+                except: 
+                    return fig
         
             ax.legend()
             ax.set_xlabel('Training steps')