Instructions to use EngrSamad/BERT-Text-Classification-Model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Scikit-learn
How to use EngrSamad/BERT-Text-Classification-Model with Scikit-learn:
from huggingface_hub import hf_hub_download import joblib model = joblib.load( hf_hub_download("EngrSamad/BERT-Text-Classification-Model", "sklearn_model.joblib") ) # only load pickle files from sources you trust # read more about it here https://skops.readthedocs.io/en/stable/persistence.html - Notebooks
- Google Colab
- Kaggle
Update config.json
Browse files- config.json +2 -3
config.json
CHANGED
|
@@ -1,10 +1,9 @@
|
|
| 1 |
{
|
| 2 |
"sklearn": {
|
| 3 |
"environment": [
|
| 4 |
-
"scikit-learn=1.
|
| 5 |
"transformers=4.22.2",
|
| 6 |
-
"torch=
|
| 7 |
-
"skorch=0.11.1.dev0"
|
| 8 |
],
|
| 9 |
"example_input": {
|
| 10 |
"data": [
|
|
|
|
| 1 |
{
|
| 2 |
"sklearn": {
|
| 3 |
"environment": [
|
| 4 |
+
"scikit-learn=1.4.1.post1",
|
| 5 |
"transformers=4.22.2",
|
| 6 |
+
"torch=2.2.1+cpu",
|
|
|
|
| 7 |
],
|
| 8 |
"example_input": {
|
| 9 |
"data": [
|