Update handler.py

handler.py

@@ -2,7 +2,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import torch
 import json
 
-class EndpointHandler:  # Asegúrate de que el nombre sea EndpointHandler
+class EndpointHandler:
     def __init__(self, model_dir):
         # Cargar el modelo y el tokenizador desde el directorio del modelo
         self.tokenizer = AutoTokenizer.from_pretrained(model_dir)
@@ -11,10 +11,10 @@ class EndpointHandler:  # Asegúrate de que el nombre sea EndpointHandler
 
     def preprocess(self, data):
         # Preprocesamiento de la entrada
-        if isinstance(data, dict) and "input_text" in data:
-            input_text = data["input_text"]
+        if isinstance(data, dict) and "inputs" in data:
+            input_text = data["inputs"]
         else:
-            raise ValueError("Esperando un diccionario con la clave 'input_text'")
+            raise ValueError("Esperando un diccionario con la clave 'inputs'")
 
         # Tokenización de la entrada
         tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding=True)
@@ -37,4 +37,3 @@ class EndpointHandler:  # Asegúrate de que el nombre sea EndpointHandler
         outputs = self.inference(tokens)
         result = self.postprocess(outputs)
         return result
-