def text_to_text_function()

in gpu-workload/t5/client-app/src/app.py [0:0]


def text_to_text_function(text: str = "Hello World", from_lang=LANG_MAP, to_lang=LANG_MAP) -> str:
  headers = {"Content-Type": "application/json"}
  payload = {"text": text, "from": from_lang, "to": to_lang}
  resp = requests.post(PREDICTION_URL, json=payload, headers=headers)
  if resp.status_code == 200:
    content = resp.json()
    return content.get("text")
  else:
    return "Oops, something went wrong!"