class_docstr: Question Answering Jax model based on TyDiQA Dataset.
code: def input_spec(self):
      return {
          "title": lit_types.TextSegment(),
          "context": lit_types.TextSegment(),
          "question": lit_types.TextSegment(),
          "answers_text": lit_types.MultiSegmentAnnotations(),
          "language": lit_types.CategoryLabel(
              required=False, vocab=question_answering.TYDI_LANG_VOCAB
          ),
      }
  
file_path: lit/lit_nlp/examples/models/tydi.py
imports: from collections.abc import Iterable
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.examples.datasets import question_answering
  import numpy
  import transformers
label_desc: Use image node_id 5 for calling the TyDiModel obj's underlying member method code with example usage: obj.input_spec() and returns: dict