def METHOD_NAME(prompt, tokens=200):
    request = {'prompt': prompt, 'max_new_tokens': tokens}
    response = requests.post(f'http://{HOST}/api/v1/generate', json=request)
    if response.status_code == 200:
        return response.json()['results'][0]['text']