File size: 805 Bytes
edfb74a 7f30e98 edfb74a 7f30e98 edfb74a 7f30e98 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
from flask import Flask, request, jsonify
from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
app = Flask(__name__)
@app.route('/translate', methods=['POST'])
def translate():
# Load the model and tokenizer for each request
tokenizer = AutoTokenizer.from_pretrained("unicamp-dl/translation-en-pt-t5")
model = AutoModelForSeq2SeqLM.from_pretrained("unicamp-dl/translation-en-pt-t5")
enpt_pipeline = pipeline('text2text-generation', model=model, tokenizer=tokenizer, max_length=512, num_beams=5)
content = request.json
text_to translate = content['text']
result = enpt_pipeline(f"translate English to Portuguese: {text_to_translate}")
return jsonify({"translated_text": result[0]["generated_text"]})
if __name__ == '__main__':
app.run(debug=True)
|