INNER CODE UNIT · Python

compose_headers

roboflow/awesome-openai-vision-api-experiments · experiments/hot-dog-not-hot-dog/app.py:71

def compose_headers(api_key: str) -> dict:
    return {
        "Content-Type": "application/json",
        "Authorization": f"Bearer {api_key}"
    }


def prompt_image(api_key: str, image: np.ndarray, prompt: str) -> str:
    headers = compose_headers(api_key=api_key)
    payload = compose_payload(image=image, prompt=prompt)
    response = requests.post(url=API_URL, headers=headers, json=payload).json()

    if 'error' in response:
        raise ValueError(response['error']['message'])
    return response['choices'][0]['message']['content']


def classify_image(api_key: str, image: np.ndarray) -> str:

View source record →