Update app.py

app.py

@@ -105,8 +105,8 @@ if (runModel=='1'):
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
         num_train_epochs=10,
-        per_device_train_batch_size=8,
-        per_device_eval_batch_size=8,
+        per_device_train_batch_size=2,
+        per_device_eval_batch_size=2,
         warmup_steps=500,
         weight_decay=0.02,
         logging_dir='./logs_' + modelNameToUse,
@@ -131,7 +131,10 @@ if (runModel=='1'):
     label_mapping = {
         0: "lastmonth",
         1: "nextweek",
-        2: "sevendays"
+        2: "sevendays",
+        3: "today",
+        4: "tomorrow",
+        5: "yesterday"
     }
 
     def evaluate_and_report_errors(model, dataloader, tokenizer):
@@ -262,7 +265,10 @@ else:
 label_mapping = {
     0: "lastmonth",
     1: "nextweek",
-    2: "sevendays"
+    2: "sevendays",
+    3: "today",
+    4: "tomorrow",
+    5: "yesterday"
 }