post_to_API.py

"""
Script to post to FastAPI instance for model inference
author: Laurent veyssier
Date: Dec. 18th 2022
"""

import requests
import json

#url = "enter heroku web app url here"
url = "https://udacity-fastapi-app.herokuapp.com/inference"


# explicit the sample to perform inference on
sample =  { 'age':50,
            'workclass':"Private", 
            'fnlgt':234721,
            'education':"Doctorate",
            'education_num':16,
            'marital_status':"Separated",
            'occupation':"Exec-managerial",
            'relationship':"Not-in-family",
            'race':"Black",
            'sex':"Female",
            'capital_gain':0,
            'capital_loss':0,
            'hours_per_week':50,
            'native_country':"United-States"
            }

data = json.dumps(sample)

# post to API and collect response
response = requests.post(url, data=data )

# display output - response will show sample details + model prediction added
print("response status code", response.status_code)
print("response content:")
print(response.json())