app.py

#10
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline

# Cargar el tokenizer y el modelo
tokenizer = AutoTokenizer.from_pretrained("PlanTL-GOB-ES/roberta-base-bne")
model = AutoModelForSeq2SeqLM.from_pretrained("PlanTL-GOB-ES/roberta-base-bne")

# Inicializar la pipeline de generación de texto
text_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)

# Generar una respuesta a una pregunta en español
question = "¿Cuál es la capital de España?"  # Ejemplo de pregunta
response = text_generator(question, max_length=50, do_sample=True)

print(response[0]['generated_text'])