HuyRemy commited on
Commit
ee4dfda
·
verified ·
1 Parent(s): ccbd678

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
README.md CHANGED
@@ -28,4 +28,14 @@ keywords = extract_keywords_model(text)
28
  # In ra màn hình những từ khóa có tần suất xuất hiện cao nhất
29
  for keyword, count in keywords.most_common():
30
  print(keyword)
 
 
 
 
 
 
 
 
 
 
31
  ```
 
28
  # In ra màn hình những từ khóa có tần suất xuất hiện cao nhất
29
  for keyword, count in keywords.most_common():
30
  print(keyword)
31
+ ```
32
+ ### Transformer
33
+ ```Python
34
+ from transformers import AutoTokenizer, AutoModel
35
+
36
+ model_path = "huyremy_extract_keywords_model.pkl"
37
+ tokenizer_path = "extract_keywords_tokenizer"
38
+
39
+ tokenizer = AutoTokenizer.from_pretrained(tokenizer_path)
40
+ model = AutoModel.from_pretrained(model_path)
41
  ```