File size: 790 Bytes
1c4dccf aef06f1 1c4dccf 37891e2 1c4dccf d2d3314 37891e2 1c4dccf d2d3314 1c4dccf |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
from fastapi import FastAPI, HTTPException
from pydantic import BaseModel
from transformers import pipeline
app = FastAPI()
# Charger le modèle de traduction
translator = pipeline("translation", model="Helsinki-NLP/opus-mt-en-fr")
class TranslationRequest(BaseModel):
text: str
@app.post("/translate/")
async def translate(request: TranslationRequest):
if not request.text:
raise HTTPException(status_code=400, detail="Le texte ne peut pas être vide.")
try:
result = translator(request.text)
return {"translated_text": result[0]['translation_text']}
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
@app.get("/")
async def root():
return {"message": "Bienvenue sur l'API de traduction avec FastAPI 🚀"}
|