class_docstr: Wraps an OWL-ViT FLAX model for convenient inference.
  
  All public methods apply to a single example and take and return Numpy arrays.
  
  Attributes:
    config: ConfigDict with model configuration.
    module: OWL-ViT Flax module.
    variables: Variable dict to be used with module.apply.
docstr: Runs the model on a dummy example to trigger compilation.
docstr_tok: ["Runs","the","model","on","a","dummy","example","to","trigger","compilation","."]
code: def warm_up(self):
      """Runs the model on a dummy example to trigger compilation."""
      image = np.zeros((128, 64, 3), dtype=np.uint8)
      queries = ("dummy",)
      query_embeddings = self.embed_text_queries(queries)
      self.get_scores(image, query_embeddings, len(queries))
  
imports: import dataclasses
  import functools
  from typing import Any, Dict, Tuple
  from flax import linen
  import jax
  import jax.numpy
  import ml_collections
  import numpy
  from scenic.model_lib.base_models import box_utils
  from scenic.projects.owl_vit.notebooks import numpy_cache
  from scipy import special
  from skimage import transform
  import tensorflow
label_desc: Use image node_id 3 for calling the Model obj's underlying member method code with example usage: obj.warm_up() without return types