Update handler.py
Browse filespadding="max_length", max_length=1000
- handler.py +1 -1
handler.py
CHANGED
@@ -12,7 +12,7 @@ class EndpointHandler:
|
|
12 |
raise ValueError("La entrada debe ser un diccionario con la clave 'inputs' y un valor válido.")
|
13 |
|
14 |
input_text = "Generate a valid JSON capturing data from this text: " + data["inputs"]
|
15 |
-
tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding=
|
16 |
if not tokens or not tokens["input_ids"]:
|
17 |
raise ValueError("Error al tokenizar el texto de entrada. Verifica el texto.")
|
18 |
return tokens
|
|
|
12 |
raise ValueError("La entrada debe ser un diccionario con la clave 'inputs' y un valor válido.")
|
13 |
|
14 |
input_text = "Generate a valid JSON capturing data from this text: " + data["inputs"]
|
15 |
+
tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1000)
|
16 |
if not tokens or not tokens["input_ids"]:
|
17 |
raise ValueError("Error al tokenizar el texto de entrada. Verifica el texto.")
|
18 |
return tokens
|