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