Update handler.py

handler.py

@@ -4,18 +4,13 @@ import joblib
 import torch
 import os
 
-# Load the label encoder
-
-from transformers import pipeline
-from sklearn.preprocessing import LabelEncoder
-import joblib
-import torch
-import os
+# Debugging: Print current directory and contents
+print("Current working directory:", os.getcwd())
+print("Contents of the directory:", os.listdir())
 
-try:
-    label_encoder = joblib.load('/repository/label_encoder.pkl')
-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 label encoder
+label_encoder = joblib.load('/repository/label_encoder.pkl')  # Use absolute path
+print("Label encoder loaded successfully.")
 
 # Load the model and tokenizer from Hugging Face
 model_name = "SCANSKY/distilbertTourism-multilingual-sentiment"
@@ -44,7 +39,7 @@ def get_average_sentiment(positive_count, negative_count, neutral_count):
     else:
         return "neutral"
 
-class Handler:
+class EndpointHandler:
     def __init__(self):
         # Model and tokenizer are loaded globally, so no need to reinitialize here
         pass
@@ -126,7 +121,7 @@ class Handler:
             "results": output
         }
 
-    def handle(self, data):
+    def __call__(self, data):
         # Main method to handle the request
         text = self.preprocess(data)
         output = self.inference(text)