added correct mapping

app.py

@@ -11,8 +11,10 @@ tokenizer = DistilBertTokenizer.from_pretrained('./fine_tuned_distilbert')
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(device)
 
-# Reverse mapping of categories to class labels
-reverse_mapping = {0: "BT1", 1: "BT2", 2: "BT3", 3: "BT4", 4: "BT5", 5: "BT6"}
+mapping = {"Remembering": 0, "Understanding": 1, "Applying": 2, "Analyzing": 3, "Evaluating": 4, "Creating": 5}
+
+# Reverse the mapping to get the class name from the index
+reverse_mapping = {v: k for k, v in mapping.items()}
 
 def predict_with_loaded_model(text):
     # Tokenize the input text