Get the prediction from the served model

predict_server.py

import requests
import json

EMOTIONS = ["angry", "disgust", "scared", "happy", "sad", "surprised", "neutral"] # Emotions corresponding to the output
data = json.dumps({"signature_name": "serving_default", "instances": img.tolist()}) # making the input in the format required for serving

headers = {"content-type": "application/json"} # specifying that the input will be in json format
json_response = requests.post('http://localhost:8501/v1/models/emotion_model:predict', data=data, headers=headers) # post request to the served model
predictions = json.loads(json_response.text)['predictions'] # getting the predictions

label = EMOTIONS[predictions.index(max(predictions))] # Matching the prediction with it's correseponding label
print(label)