tests.py

from handler import EndpointHandler

# init handler
response_model_handler = EndpointHandler()

# prepare sample payload
payload = {"instruction": "What are some ways to stay energized throughout the day?",
           "response": "Drink lots of coffee!"}

# test the handler
pred=response_model_handler(payload)

print(pred)