Update app.py

app.py

@@ -128,14 +128,35 @@ if (runModel=='1'):
     # Evaluate the model
     trainer.evaluate()
 
+    #label_mapping = {
+        #0: "lastmonth",
+        #1: "nextweek",
+        #2: "sevendays",
+        #3: "today",
+        #4: "tomorrow",
+        #5: "yesterday"
+    #}
     label_mapping = {
-        0: "lastmonth",
-        1: "nextweek",
-        2: "sevendays",
-        3: "today",
-        4: "tomorrow",
-        5: "yesterday"
+        0: "fivedays",
+        1: "fiveyears",
+        2: "fourdays",
+        3: "one-year",
+        4: "oneday",
+        5: "onemonth",
+        6: "sevendays",
+        7: "sixdays",
+        8: "tenyears",
+        9: "threedays",
+        10: "threeweeks",
+        11: "threemonths",
+        12: "thismonth",
+        13: "thisweek",
+        14: "twodays",
+        15: "twomonths",
+        16: "twoweeks",
+
     }
+    
 
     def evaluate_and_report_errors(model, dataloader, tokenizer):
         model.eval()
@@ -262,15 +283,36 @@ else:
     #tokenizer = AutoTokenizer.from_pretrained(tokenizer_save_path)
 
 #Define the label mappings (this must match the mapping used during training)
+
+#label_mapping = {
+    #0: "lastmonth",
+    #1: "nextweek",
+    #2: "sevendays",
+    #3: "today",
+    #4: "tomorrow",
+    #5: "yesterday"
+#}
+
 label_mapping = {
-    0: "lastmonth",
-    1: "nextweek",
-    2: "sevendays",
-    3: "today",
-    4: "tomorrow",
-    5: "yesterday"
-}
+    0: "fivedays",
+    1: "fiveyears",
+    2: "fourdays",
+    3: "one-year",
+    4: "oneday",
+    5: "onemonth",
+    6: "sevendays",
+    7: "sixdays",
+    8: "tenyears",
+    9: "threedays",
+    10: "threeweeks",
+    11: "threemonths",
+    12: "thismonth",
+    13: "thisweek",
+    14: "twodays",
+    15: "twomonths",
+    16: "twoweeks",
 
+}
 
 #Function to classify user input
 def classifyTimeFrame(user_input):