Instructions to use Aalaa/Fine_tuned_Vit_trash_classification with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use Aalaa/Fine_tuned_Vit_trash_classification with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="Aalaa/Fine_tuned_Vit_trash_classification") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# pip install -U transformers accelerate # Load model directly from transformers import AutoImageProcessor, AutoModelForImageClassification processor = AutoImageProcessor.from_pretrained("Aalaa/Fine_tuned_Vit_trash_classification") model = AutoModelForImageClassification.from_pretrained("Aalaa/Fine_tuned_Vit_trash_classification", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -30,7 +30,7 @@ The dataset used consist of spans six classes: glass, paper, cardboard, plastic
|
|
| 30 |
* 137 trash
|
| 31 |
## Fine_tuned Notebook
|
| 32 |
|
| 33 |
-
This notebook outlines the steps from preparing the data in the Vit-acceptable format to training the model [Notebook]
|
| 34 |
|
| 35 |
### How to use
|
| 36 |
|
|
|
|
| 30 |
* 137 trash
|
| 31 |
## Fine_tuned Notebook
|
| 32 |
|
| 33 |
+
This notebook outlines the steps from preparing the data in the Vit-acceptable format to training the model [Notebook](https://colab.research.google.com/drive/1RbmRPJ9bFLA_qK9RGgPoHZRnUTy_md5O?usp=sharing)
|
| 34 |
|
| 35 |
### How to use
|
| 36 |
|