Instructions to use daxa-ai/pebblo-classifier with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use daxa-ai/pebblo-classifier with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="daxa-ai/pebblo-classifier")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("daxa-ai/pebblo-classifier") model = AutoModelForSequenceClassification.from_pretrained("daxa-ai/pebblo-classifier") - Notebooks
- Google Colab
- Kaggle
Custom inference code for SageMaker deployment
#9
by raj-daxa - opened
No description provided.
Custom inference code for SageMaker deployment:
The inference.py script is necessary for SageMaker deployment to return all the topics in the response. Otherwise, it returns only the single topic with the highest score.
raj-daxa changed pull request status to open
sridhar-cd changed pull request status to merged
LGTM