File size: 300 Bytes
fc58506
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
from fastapi import FastAPI
from transformers import pipeline

app = FastAPI()
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-id")

@app.post("/translate")
async def translate(text: str):
    result = translator(text)
    return {"translated_text": result[0]["translation_text"]}