Update handler.py

handler.py

@@ -17,10 +17,15 @@ class EndpointHandler:
         base_model = AutoModelForSequenceClassification.from_pretrained(model_name)
         self.model = PeftModel.from_pretrained(base_model, model_name)
 
-    def __call__(self, input_text):
+    def __call__(self, input_data):
         """
         Perform inference on the input text and return predictions.
         """
+                # Extract the URL from the input_data dictionary
+        if 'inputs' in input_data:
+            input_text = input_data['inputs']  # Expecting a single URL as a string
+        else:
+            raise ValueError("Input data must contain the 'inputs' key with a URL.")
         # Tokenize input
         inputs = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding=True)