Update handler.py

handler.py

@@ -6,11 +6,16 @@ import os
 
 # Load the label encoder
 
+from transformers import pipeline
+from sklearn.preprocessing import LabelEncoder
+import joblib
+import torch
+import os
+
 try:
     label_encoder = joblib.load('/repository/label_encoder.pkl')
-
-except error:
-    print("COULD NOT FIND LABEL_ENCODER. HERE'S THE CWD -->"+string(os.getcwd()))
+except Exception as e:
+    raise FileNotFoundError("Label encoder file not found: /repository/label_encoder.pkl in cwd " os.getcwd()) #Raise a file not found error.
 
 # Load the model and tokenizer from Hugging Face
 model_name = "SCANSKY/distilbertTourism-multilingual-sentiment"