Update app.py

app.py

@@ -128,33 +128,13 @@ 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: "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",
-
+        0: "lastmonth",
+        1: "nextweek",
+        2: "sevendays",
+        3: "today",
+        4: "tomorrow",
+        5: "yesterday"
     }
     
 
@@ -284,34 +264,13 @@ else:
 
 #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: "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",
-
+    0: "lastmonth",
+    1: "nextweek",
+    2: "sevendays",
+    3: "today",
+    4: "tomorrow",
+    5: "yesterday"
 }
 
 #Function to classify user input