mayapapaya commited on
Commit
b053c99
1 Parent(s): 6be7cb3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
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
- ### Training Data
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
+ ```