mayapapaya
commited on
Commit
•
b053c99
1
Parent(s):
6be7cb3
Update README.md
Browse files
README.md
CHANGED
@@ -11,6 +11,17 @@ This model is meant to extract keywords from text.
|
|
11 |
|
12 |
This model is a fine-tuned version of the [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) model.
|
13 |
|
14 |
-
|
15 |
|
16 |
Trained on [51la5/keyword-extraction](https://huggingface.co/datasets/51la5/keyword-extraction) from HuggingFace Hub.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
11 |
|
12 |
This model is a fine-tuned version of the [distilbert-base-uncased-finetuned-sst-2-english](https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english) model.
|
13 |
|
14 |
+
## Training Data
|
15 |
|
16 |
Trained on [51la5/keyword-extraction](https://huggingface.co/datasets/51la5/keyword-extraction) from HuggingFace Hub.
|
17 |
+
|
18 |
+
## How to Get Started with the Model
|
19 |
+
|
20 |
+
Note: model inputs were tokenized using distilbert-base-uncased tokenizer
|
21 |
+
|
22 |
+
```
|
23 |
+
from transformers import AutoTokenizer, AutoModelForSequenceClassification
|
24 |
+
|
25 |
+
tokenizer = AutoTokenizer.from_pretrained("distilbert-base-uncased")
|
26 |
+
model = AutoModelForSequenceClassification.from_pretrained("mayapapaya/Keyword-Extractor")
|
27 |
+
```
|