Update app.py

app.py

@@ -11,6 +11,7 @@ from transformers import (
     TrainingArguments,
     Trainer,
     pipeline,
+    AutoModelForQuestionAnswering, 
 )
 import evaluate # type: ignore
 from huggingface_hub import login
@@ -48,7 +49,7 @@ def detect_language(text):
         return "en"  # Si no se puede detectar, se asume inglés
 
 # Generación de texto (si se solicita un cuento, por ejemplo)
-def generate_text(prompt, max_length=100):
+def generate_text(prompt, max_length=2048):
     output = generator(prompt, max_length=int(max_length), num_return_sequences=1)[0]["generated_text"]
     return output
 
@@ -82,11 +83,11 @@ def process_input(user_input):
     elif "cuento" in user_input.lower():
         # Si el usuario solicita un cuento
         prompt = "Había una vez, en un reino lejano..."
-        return generate_text(prompt, max_length=200)
+        return generate_text(prompt, max_length=500)
     elif "emoción" in user_input.lower():
         # Si es una solicitud de apoyo emocional
         context = "Dom: Psicología | Estilo: Apoyo emocional. ¿Cómo te sientes hoy?"
-        return generate_text(context + " Estoy aquí para apoyarte", max_length=100)
+        return generate_text(context + " Estoy aquí para apoyarte", max_length=2048)
     else:
         return "No entiendo la solicitud, por favor intenta preguntar algo más específico."