Update app.py

app.py

@@ -32,6 +32,17 @@ class BertModule(nn.Module):
         pred = self.bert(**kwargs)
         return pred.logits
 
+MAX_EPOCHS = 5
+BATCH_SIZE = 12
+
+num_training_steps = MAX_EPOCHS * (len(X_train) // BATCH_SIZE + 1)
+
+def lr_schedule(current_step):
+    factor = float(num_training_steps - current_step) / float(max(1, num_training_steps))
+    assert factor > 0
+    return factor
+
+
 
 with open('learner.bin', 'rb') as f:
     learner = pickle.load(f)