from transformers import pipeline
def responder_mensaje(mensaje):
"""
Función que recibe un mensaje y devuelve una respuesta razonada.
"""
chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium")
respuesta = chatbot(mensaje, max_length=1000, do_sample=True)
return respuesta[0]["generated_text"]
Ejemplo de uso
txt_usuario = input("Escribe un mensaje: ")
print(responder_mensaje(txt_usuario))
