File size: 448 Bytes
0026ff3 f38a916 0026ff3 f38a916 0026ff3 f38a916 0026ff3 f38a916 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
from handler import EndpointHandler
def test_handler():
# Initialize the handler
handler = EndpointHandler()
# Test with a simple prompt
test_input = {
"inputs": "Explain quantum computing in simple terms"
}
# Get the response
response = handler(test_input)
print("Input:", test_input["inputs"])
print("Output:", response["generated_text"])
if __name__ == "__main__":
test_handler() |