in src/mozmlops/templates/template_ray_serve.py [0:0]
def translate(self, translate_request: TranslateRequest) -> str:
# Run inference
model_output = self.model(translate_request.text)
# Post-process output to return only the translation text
translation = model_output[0]["translation_text"]
return translation