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) |