Update handler.py

handler.py

@@ -8,18 +8,29 @@ class EndpointHandler:
         self.vectorizer = joblib.load(os.path.join(model_dir, 'vectorizer.joblib'))
         self.model = joblib.load(os.path.join(model_dir, 'logistic_classifier.joblib'))
 
+        # Check if the vectorizer is fitted
+        if not hasattr(self.vectorizer, 'vocabulary_'):
+            raise ValueError("The vectorizer is not fitted. Ensure the vectorizer is trained and saved correctly.")
+
+        # Check if the model is fitted
+        if not hasattr(self.model, 'classes_'):
+            raise ValueError("The model is not fitted. Ensure the model is trained and saved correctly.")
+
+        print("Vectorizer and model loaded successfully.")
+
         # Verify that the tokenizer configuration is correct
         with open(os.path.join(model_dir, "tokenizer.json"), "r") as file:
             tokenizer_config = json.load(file)
         if tokenizer_config['tokenizer'] != 'split':
             raise ValueError("Tokenizer configuration does not match the expected tokenizer.")
 
+        print("Tokenizer configuration verified.")
+
     def predict_rating(self, review):
         review_tfidf = self.vectorizer.transform([review])
         predicted_rating = self.model.predict(review_tfidf)[0]
         return int(predicted_rating)
 
-    # __call__ method should be part of the class
     def __call__(self, inputs):
         try:
             # Parse the input JSON string
@@ -41,16 +52,13 @@ class EndpointHandler:
             if not isinstance(review, str) or not review.strip():
                 return json.dumps({"error": "Review must be a non-empty string."})
 
-            # Perform prediction
             predicted_rating = self.predict_rating(review)
 
-            # Prepare response
             response = {
                 "review": review,
                 "predicted_rating": predicted_rating
             }
 
-            # Return JSON response
             return json.dumps(response)
 
         except json.JSONDecodeError:
@@ -58,3 +66,4 @@ class EndpointHandler:
 
         except Exception as e:
             return json.dumps({"error": str(e)})
+