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