mrm8488
/

vit-base-patch16-224_finetuned-kvasirv2-colonoscopy

Image Classification

Inference Endpoints

Model card Files Files and versions Metrics Training metrics Community

mrm8488 commited on Sep 7, 2021

Commit

d3b4693

•

1 Parent(s): e38f17b

Update README.md

Files changed (1) hide show

README.md +4 -0

README.md CHANGED Viewed

@@ -42,6 +42,10 @@ classifier = VisionClassifierInference(
     feature_extractor = ViTFeatureExtractor.from_pretrained(path),
     model = ViTForImageClassification.from_pretrained(path),
 )
 ```
 > Disclaimer: This model was trained for research only

     feature_extractor = ViTFeatureExtractor.from_pretrained(path),
     model = ViTForImageClassification.from_pretrained(path),
 )
+img  = "Your image path"
+label = classifier.predict(img_path=img)
+print("Predicted class:", label)
 ```
 > Disclaimer: This model was trained for research only