Instructions to use gohjiayi/suicidal-bert with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use gohjiayi/suicidal-bert with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-classification", model="gohjiayi/suicidal-bert")# Load model directly from transformers import AutoTokenizer, AutoModelForSequenceClassification tokenizer = AutoTokenizer.from_pretrained("gohjiayi/suicidal-bert") model = AutoModelForSequenceClassification.from_pretrained("gohjiayi/suicidal-bert") - Notebooks
- Google Colab
- Kaggle
How can I actually use your model?
#1
by EvelinaAlex - opened
If you would like to infer on Python, you can refer to this file, more specifically the check_intent() function. Make sure you load the tokenizer and model beforehand.
https://github.com/gohjiayi/suicidal-text-detection/blob/main/chatbot.ipynb
