Update app.py

app.py

@@ -10,6 +10,17 @@ from huggingface_hub import login
 token = os.getenv("hf_token")
 login(token=token)
 
+batch_size = 96
+n_epochs = 2
+base_LM_model = "roberta-base"
+max_seq_len = 512
+learning_rate = 3e-5
+lr_schedule = LinearWarmup
+warmup_proportion = 0.2
+doc_stride=128
+max_query_length=64
+
+