When I tried asking the same question as provided in AWS Sagemaker deployment script, it gave:
And then checked with { "inputs": "def sum(a,b):" } And it returned: { "generated_text": "def sum(a,b):\n return a+b\n\ndef diff(a,b):\n return a-b\n" } How do I get it to give complete code or make it to chat?