Update app.py

app.py

@@ -104,7 +104,7 @@ if (runModel=='1'):
     # Create an instance of the custom loss function
     training_args = TrainingArguments(
         output_dir='./results_' + modelNameToUse,
-        num_train_epochs=1,
+        num_train_epochs=25,
         per_device_train_batch_size=2,
         per_device_eval_batch_size=2,
         warmup_steps=500,
@@ -131,10 +131,7 @@ if (runModel=='1'):
     label_mapping = {
         0: "lastmonth",
         1: "nextweek",
-        2: "sevendays",
-        3: "today",
-        4: "tomorrow",
-        5: "yesterday"
+        2: "sevendays"
         
     }
 
@@ -255,10 +252,7 @@ else:
 label_mapping = {
     0: "lastmonth",
     1: "nextweek",
-    2: "sevendays",
-    3: "today",
-    4: "tomorrow",
-    5: "yesterday"
+    2: "sevendays"
 }