How to use marlonbino/ner-distilbert-base-cased with Scikit-learn:
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("marlonbino/ner-distilbert-base-cased", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html
How to use marlonbino/ner-distilbert-base-cased with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="marlonbino/ner-distilbert-base-cased")
# Load model directly from transformers import AutoTokenizer, AutoModelForTokenClassification tokenizer = AutoTokenizer.from_pretrained("marlonbino/ner-distilbert-base-cased") model = AutoModelForTokenClassification.from_pretrained("marlonbino/ner-distilbert-base-cased", device_map="auto")
The community tab is the place to discuss and collaborate with the HF community!
