English tiny_random_levitforimageclassificationwithteacher ViTForImageClassification from hf-tiny-model-private

Description

Pretrained ViTForImageClassification model, adapted from Hugging Face and curated to provide scalability and production-readiness using Spark NLP.tiny_random_levitforimageclassificationwithteacher is a English model originally trained by hf-tiny-model-private.

Download Copy S3 URI

How to use

		
		

image_assembler = ImageAssembler()\
  .setInputCol("image")\
  .setOutputCol("image_assembler")

imageClassifier = ViTForImageClassification.pretrained(""tiny_random_levitforimageclassificationwithteacher","en")\
  .setInputCols("image_assembler")\
  .setOutputCol("class")

pipeline = Pipeline(stages=[
  image_assembler,
  imageClassifier,
])

pipelineModel = pipeline.fit(imageDF)

pipelineDF = pipelineModel.transform(imageDF)

val imageAssembler = new ImageAssembler()
    .setInputCol("image")
    .setOutputCol("image_assembler")

val imageClassifier =  ViTForImageClassification.pretrained("tiny_random_levitforimageclassificationwithteacher","en") 
    .setInputCols("image_assembler") 
    .setOutputCol("class") 

val pipeline = new Pipeline().setStages(Array(imageAssembler, imageClassifier))

val pipelineModel = pipeline.fit(imageDF)

val pipelineDF = pipelineModel.transform(imageDF)

Model Information

Model Name:	tiny_random_levitforimageclassificationwithteacher
Compatibility:	Spark NLP 5.5.1+
License:	Open Source
Edition:	Official
Input Labels:	[image_assembler]
Output Labels:	[label]
Language:	en
Size:	26.0 MB

References

https://huggingface.co/hf-tiny-model-private/tiny-random-LevitForImageClassificationWithTeacher

PREVIOUSEnglish tiny_random_debertav2formaskedlm_pipeline pipeline BertEmbeddings from hf-tiny-model-private

NEXTEnglish tiny_random_levitforimageclassificationwithteacher_pipeline pipeline ViTForImageClassification from hf-tiny-model-private