Update app.py

app.py

@@ -25,6 +25,8 @@ def plot_loss_rates(df, type):
     x = np.linspace(0, 1, 1000)
     loss_rates = []
     labels = ['32', '64', '128', '256', '512', '1024']
+        #drop the column step
+    df = df.drop(columns=['Step'])
     for col in df.columns:
         y = df[col].values.dropna()
         f = interp1d(np.linspace(0, 1, len(y)), y)