python
from flask import Flask, request, jsonify
from transformers import T5Tokenizer, T5ForConditionalGeneration
app = Flask(__name__)
加载预训练模型和tokenizer
model = T5ForConditionalGeneration.from_pretrained('t5small')
tokenizer = T5Tokenizer.from_pretrained('t5small')
@app.route('/generate', methods=['POST'])
def generate_content():
data = request.json
if 'prompt' not in data:
return jsonify({'error': 'Missing prompt'}), 400
prompt = data['prompt']
input_ids = tokenizer.encode(generate a story about + prompt, return_tensors=pt)
output = model.generate(input_ids, max_length=150)
generated_text = tokenizer.decode(output[0], skip_special_tokens=True)