name: 'vit-embedding'
labels: 
  recommended_framework: pytorch1.2.0
  class: vit-embedding
  others: vit
operator: 'towhee/vit-embedding'
init:
  model_name: str
call:
  input:
    image: towhee.types.Image
  output:
    feature_vector: numpy.ndarray