Help me to convert Huggingface TFGPT2LMHeadModel to TFLITE

Job ID: 35706716

Budget: ₹600 – ₹1,500 INR

I want to create an android app which uses TFLITE model to predict next sentences but i see the output shape is complete different . For more info refer :

https://stackoverflow.com/questions/75050453/tfgpt2lmheadmodel-to-tflite-changes-the-input-and-output-shape

How can we fix the same and import the same to the android project . Need an expert on Huggingface