Update app.py

app.py

@@ -35,7 +35,7 @@ class BertModule(nn.Module):
 MAX_EPOCHS = 5
 BATCH_SIZE = 12
 
-num_training_steps = MAX_EPOCHS * (len(X_train) // BATCH_SIZE + 1)
+num_training_steps = MAX_EPOCHS * (584 // BATCH_SIZE + 1)
 
 def lr_schedule(current_step):
     factor = float(num_training_steps - current_step) / float(max(1, num_training_steps))