jla25 commited on
Commit
2433496
·
verified ·
1 Parent(s): 5888de7

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -5
handler.py CHANGED
@@ -13,11 +13,7 @@ class EndpointHandler:
13
  raise ValueError("La entrada debe ser un diccionario con la clave 'inputs' y un valor válido.")
14
 
15
  # Prompt personalizado para guiar al modelo
16
- input_text = (
17
- f"""
18
- ### Procesa el siguiente texto y genera un JSON válido:
19
- "{data['inputs']}"
20
- """)
21
  # Imprimir el texto generado para el prompt
22
  print(f"Prompt generado para el modelo: {input_text}")
23
  tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1024)
 
13
  raise ValueError("La entrada debe ser un diccionario con la clave 'inputs' y un valor válido.")
14
 
15
  # Prompt personalizado para guiar al modelo
16
+ input_text = ({data['inputs']})
 
 
 
 
17
  # Imprimir el texto generado para el prompt
18
  print(f"Prompt generado para el modelo: {input_text}")
19
  tokens = self.tokenizer(input_text, return_tensors="pt", truncation=True, padding="max_length", max_length=1024)