Instructions to use michotic/vit_cancer_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use michotic/vit_cancer_model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="michotic/vit_cancer_model") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("michotic/vit_cancer_model") model = AutoModelForImageClassification.from_pretrained("michotic/vit_cancer_model") - Notebooks
- Google Colab
- Kaggle
YAML Metadata Warning:empty or missing yaml metadata in repo card
Check out the documentation for more information.
Fine-tuned ViT model trained to predict types of lung cancer / no-cancer from CT-scans of the torso.
- Downloads last month
- 5