Instructions to use dejanai/LinkBERT with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use dejanai/LinkBERT with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("token-classification", model="dejanai/LinkBERT")# pip install -U transformers accelerate # Load model directly from transformers import AutoTokenizer, AutoModelForMaskedLM tokenizer = AutoTokenizer.from_pretrained("dejanai/LinkBERT") model = AutoModelForMaskedLM.from_pretrained("dejanai/LinkBERT", device_map="auto") - Notebooks
- Google Colab
- Kaggle
Update handler.py
Browse files- handler.py +1 -0
handler.py
CHANGED
|
@@ -50,3 +50,4 @@ class EndpointHandler:
|
|
| 50 |
|
| 51 |
# Note: Ensure the path "dejanseo/LinkBERT" is correctly pointing to your model's location
|
| 52 |
# If the model is locally saved, adjust the path accordingly
|
|
|
|
|
|
| 50 |
|
| 51 |
# Note: Ensure the path "dejanseo/LinkBERT" is correctly pointing to your model's location
|
| 52 |
# If the model is locally saved, adjust the path accordingly
|
| 53 |
+
|