Update app.py

app.py

@@ -22,12 +22,13 @@ def generate_response(prompt):
 
     outputs = model.generate(
         inputs.input_ids,
-        max_new_tokens=50,  # 🔹 En vez de max_length (mejor control de generación)
+        attention_mask=inputs.attention_mask,  # 🔹 Asegura que el modelo procese correctamente la entrada
+        max_new_tokens=50,  # 🔹 Control preciso sobre la generación
         do_sample=True,
-        temperature=0.45,  # 🔹 Menos aleatoriedad, más coherencia
-        top_p=0.9,  # 🔹 Más controlado
-        repetition_penalty=1.1,  # 🔹 Evita repeticiones
-        early_stopping=True,
+        temperature=0.45,  # 🔹 Reduce la aleatoriedad sin hacer respuestas robóticas
+        top_p=0.9,  # 🔹 Más fluidez y control en la elección de palabras
+        repetition_penalty=1.1,  # 🔹 Evita repeticiones innecesarias
+        early_stopping=False,  # 🔹 Ajustado para evitar la advertencia en Hugging Face
     )
 
     return tokenizer.decode(outputs[0], skip_special_tokens=True)