img_seg.py

import requests

model_id = "nvidia/segformer-b0-finetuned-ade-512-512"
api_token = # get yours at hf.co/settings/tokens

headers = {"Authorization": f"Bearer {api_token}"}



API_URL = f"https://api-inference.huggingface.co/models/{model_id}"

def query(data, model_id, api_token):
    with open(filename, "rb") as f:
        data = f.read()
    response = requests.post(API_URL, headers=headers, data=data)
    return response.json()


filename = 'SCR-tfa.jpeg'

data = query(filename, model_id, api_token)

print(data)