Spaces:
Sleeping
Sleeping
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.7, max_new_tokens=
|
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)
|