Garabatos commited on
Commit
707d46d
·
1 Parent(s): 1dc2772

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -24,8 +24,12 @@ def chat(msg: Message):
24
  input_text = msg.text # Texto de entrada
25
  print(f"Mensaje recibido: {input_text}")
26
 
 
 
 
 
27
  # Codificar el texto de entrada y agregar el token de fin de secuencia
28
- inputs = tokenizer.encode(input_text + tokenizer.eos_token, return_tensors="pt")
29
 
30
  # Generar la respuesta
31
  response_ids = model.generate(inputs,
 
24
  input_text = msg.text # Texto de entrada
25
  print(f"Mensaje recibido: {input_text}")
26
 
27
+ messages = [
28
+ {"role": "user", "content": "{input_text}"},
29
+ ]
30
+
31
  # Codificar el texto de entrada y agregar el token de fin de secuencia
32
+ inputs = tokenizer.encode(messages + tokenizer.eos_token, return_tensors="pt")
33
 
34
  # Generar la respuesta
35
  response_ids = model.generate(inputs,