Update handler.py

handler.py

@@ -1,49 +1,31 @@
-import joblib
-import os
-import json
+def __call__(self, inputs):
+    try:
+        inputs_dict = json.loads(inputs)
 
-class EndpointHandler:
-    def __init__(self, model_dir):
-        self.model_dir = model_dir
-        self.vectorizer = joblib.load(os.path.join(model_dir, 'vectorizer.joblib'))
-        self.model = joblib.load(os.path.join(model_dir, 'logistic_classifier.joblib'))
+        if 'inputs' not in inputs_dict:
+            return json.dumps({"error": "No valid inputs provided"})
 
-        # 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.")
+        inputs_data = inputs_dict['inputs']
 
-    def predict_rating(self, review):
-        review_tfidf = self.vectorizer.transform([review])
-        predicted_rating = self.model.predict(review_tfidf)[0]
-        return int(predicted_rating)
+        if 'review' not in inputs_data:
+            return json.dumps({"error": "No valid review provided"})
 
-    def __call__(self, inputs):
-        try:
-            inputs_dict = json.loads(inputs)
+        review = inputs_data['review']
 
-            if 'review' not in inputs_dict:
-                return json.dumps({"error": "No valid review provided"})
+        if not review:
+            return json.dumps({"error": "No valid review provided"})
 
-            review = inputs_dict['review']
+        predicted_rating = self.predict_rating(review)
 
-            if not review:
-                return json.dumps({"error": "No valid review provided"})
+        response = {
+            "review": review,
+            "predicted_rating": predicted_rating
+        }
 
-            predicted_rating = self.predict_rating(review)
-
-            response = {
-                "review": review,
-                "predicted_rating": predicted_rating
-            }
-
-            return json.dumps(response)
-
-        except json.JSONDecodeError:
-            return json.dumps({"error": "Invalid JSON format"})
-
-        except Exception as e:
-            return json.dumps({"error": str(e)})
+        return json.dumps(response)
 
+    except json.JSONDecodeError:
+        return json.dumps({"error": "Invalid JSON format"})
 
+    except Exception as e:
+        return json.dumps({"error": str(e)})