jla25 commited on
Commit
4722f73
·
verified ·
1 Parent(s): 05da434

Update handler.py

Browse files

padding="max_length", max_length=1000

Files changed (1) hide show
  1. 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=True)
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