Update app.py

app.py

@@ -24,17 +24,17 @@ 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
+    #drop the column step
     df = df.drop(columns=['Step'])
     for col in df.columns:
-        y = df[col].dropna().values
+        y = df[col].dropna().astype('float', errors = 'ignore').dropna().values
         f = interp1d(np.linspace(0, 1, len(y)), y)
         loss_rates.append(f(x))
     fig, ax = plt.subplots()
     for i, loss_rate in enumerate(loss_rates):
         ax.plot(x, loss_rate, label=labels[i])
     ax.legend()
-    ax.set_title(f'Loss rates for {type} model')
+    ax.set_title(f'Loss rates for a {type} parameter model')
     ax.set_xlabel('Training steps')
     ax.set_ylabel('Loss rate')
     return fig