Shanav12/sports_ball_dataset
Viewer • Updated • 220 • 21
How to use raffel-22/my-custom-model with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("image-classification", model="raffel-22/my-custom-model")
pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png") # Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("raffel-22/my-custom-model", dtype="auto")It was a task from the AI Engineer boot camp, if you want to improve this repo, you can contact me, see my code to build this repo via this link :
https://colab.research.google.com/drive/1Nmvnyma8pFovz_WsucC05dJhR5T9DR_I?usp=sharing