Error while running custom models on Google Coral EdgeTPU, "list index out of range"

Question

I tried running a custom model on the Google Coral EdgeTPU, but it is causing the following error:

  File "detect.py", line 102, in <module>
    main()
  File "detect.py", line 74, in main
    objs = get_objects(interpreter, args.threshold)[:args.top_k]
  File "/usr/lib/python3/dist-packages/pycoral/adapters/detect.py", line 214, in get_objects
    elif common.output_tensor(interpreter, 3).size == 1:
  File "/usr/lib/python3/dist-packages/pycoral/adapters/common.py", line 29, in output_tensor
    return interpreter.tensor(interpreter.get_output_details()[i]['index'])()
IndexError: list index out of range

The default model mobilenet_ssd_v2_coco_quant_postprocess_edgetpu.tflite works perfectly

I tried a running a simple 2 layer CNN model and a transfer learning based mobilenet model, but both showed the above errors. The thread about running YOLOv5 didn't help much. Any help would be great. Thanks a lot