jla25 commited on
Commit
4cb23e7
·
verified ·
1 Parent(s): ec5fdca

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -22,7 +22,7 @@ class EndpointHandler:
22
  # Prompt personalizado para guiar al modelo
23
  input_text = f"Generate a valid JSON capturing data from this text:{data['inputs']}"
24
  print(f"Prompt generado para el modelo: {input_text}")
25
-
26
  tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1024)
27
  return tokens
28
 
 
22
  # Prompt personalizado para guiar al modelo
23
  input_text = f"Generate a valid JSON capturing data from this text:{data['inputs']}"
24
  print(f"Prompt generado para el modelo: {input_text}")
25
+ input_text = input_text.encode("utf-8").decode("utf-8")
26
  tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1024)
27
  return tokens
28