Spaces:
Sleeping
Sleeping
JaphetHernandez
commited on
Update app.py
Browse files
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.
|
23 |
"""Funci贸n de generaci贸n de texto con el modelo."""
|
24 |
input_ids = tokenizer.encode(input_text, return_tensors='pt').to(model.device)
|
25 |
|
|
|
19 |
# Definir longitud m谩xima de tokens
|
20 |
MAX_INPUT_TOKEN_LENGTH = 4096
|
21 |
|
22 |
+
def generate_response(input_text, temperature=0.2, max_new_tokens=100):
|
23 |
"""Funci贸n de generaci贸n de texto con el modelo."""
|
24 |
input_ids = tokenizer.encode(input_text, return_tensors='pt').to(model.device)
|
25 |
|