JaphetHernandez commited on
Commit
47ec4fb
verified
1 Parent(s): 3cbbebc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ tokenizer.pad_token = tokenizer.eos_token
19
  # Definir longitud m谩xima de tokens
20
  MAX_INPUT_TOKEN_LENGTH = 4096
21
 
22
- def generate_response(input_text, temperature=0.7, max_new_tokens=1000):
23
  """Funci贸n de generaci贸n de texto con el modelo."""
24
  input_ids = tokenizer.encode(input_text, return_tensors='pt')
25
  input_ids = input_ids.to(model.device)
 
19
  # Definir longitud m谩xima de tokens
20
  MAX_INPUT_TOKEN_LENGTH = 4096
21
 
22
+ def generate_response(input_text, temperature=0.7, max_new_tokens=50):
23
  """Funci贸n de generaci贸n de texto con el modelo."""
24
  input_ids = tokenizer.encode(input_text, return_tensors='pt')
25
  input_ids = input_ids.to(model.device)