mateoluksenberg commited on
Commit
252bfd7
·
verified ·
1 Parent(s): a823ed0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -242,8 +242,7 @@ def simple_chat(message: dict, temperature: float = 0.8, max_length: int = 4096,
242
  else:
243
  conversation.append({"role": "user", "content": message['text']})
244
 
245
- # Convierte solo el contenido relevante a IDs
246
- input_ids = tokenizer(conversation[-1]['content'], return_tensors="pt").to(model.device)
247
 
248
  generate_kwargs = dict(
249
  max_length=max_length,
 
242
  else:
243
  conversation.append({"role": "user", "content": message['text']})
244
 
245
+ input_ids = tokenizer(input_text, return_tensors="pt").to(model.device)
 
246
 
247
  generate_kwargs = dict(
248
  max_length=max_length,