Translator / app.py
feliksius's picture
Create app.py
fc58506 verified
raw
history blame
300 Bytes
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"]}