Hi @Rohit,
You could also set the pipeline like:
Screen Shot 2021-01-01 at 9.52.27 AM2150×246 50.7 KB
and use pipeline([code]) to generate description for the input code.
Here is one Colab example: https://github.com/agemagician/CodeTrans/blob/main/prediction/multitask/fine-tuning/function%20documentation%20generation/java/base_model.ipynb 18