Update app.py

app.py

@@ -101,18 +101,22 @@ def conversar_con_colmena(mensaje):
     
     with torch.no_grad():
         output = model.generate(
-            **inputs,
-            max_length=200,
-            do_sample=True,
-            top_p=0.9
+    **inputs,
+    max_length=150,  # Reduce el tamaño máximo para evitar que se quede colgado
+    do_sample=True,   # Sampling activado para más variabilidad
+    top_k=50,         # Controla la aleatoriedad para evitar salidas vacías
+    temperature=0.7,  # Ajusta la creatividad de la respuesta
+    pad_token_id=tokenizer.eos_token_id  # Evita errores de padding
+)
+
         )
 
-    respuesta = tokenizer.decode(output[0], skip_special_tokens=True)
-    print(f"✅ Respuesta generada por la IA: {respuesta}")  # 👈 Agrega este print para ver la respuesta generada
-    
-    memory.save_context({"input": mensaje}, {"output": respuesta})
-    
-    return respuesta
+    respuesta = tokenizer.decode(output[0], skip_special_tokens=True).strip()
+
+    print(f"✅ Respuesta generada por la IA: '{respuesta}'")  # 👈 Para ver si está vacío
+
+    if not respuesta:
+    return "🤖 No pude generar una respuesta. Inténtalo de nuevo con otra pregunta."
 
 
 iface = gr.Interface(