Grader Functions_ failed tests for create_model

The response is shown as below:

Failed test case: your model could not be used for inference. Details shown in ‘got’ value below:.
no exceptions,
but got:
Asked to retrieve element 0, but the Sequence has length 0.

My function for model creation is as below:

Can you please PM me the notebook you use in pdf form by clicking on my name
Thanks and Regards,
Mayank Ghogale