Instructions to use DataScienceProject/Vit with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use DataScienceProject/Vit with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="DataScienceProject/Vit") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")# Load model directly from transformers import AutoModel model = AutoModel.from_pretrained("DataScienceProject/Vit", dtype="auto") - Notebooks
- Google Colab
- Kaggle
Update requirements.txt
Browse files- requirements.txt +4 -3
requirements.txt
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
-
|
| 2 |
-
|
| 3 |
-
|
|
|
|
| 4 |
Pillow
|
| 5 |
pandas
|
| 6 |
scikit-learn
|
|
|
|
| 1 |
+
# Requires Python 3.10.14
|
| 2 |
+
torch==2.3.0+cu121
|
| 3 |
+
torchvision==0.18.0
|
| 4 |
+
transformers==4.40.2
|
| 5 |
Pillow
|
| 6 |
pandas
|
| 7 |
scikit-learn
|