AlexTANG-JX commited on
Commit
ffadfb5
·
verified ·
1 Parent(s): fa3af31

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -37,7 +37,7 @@ python inference.py
37
  from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
38
 
39
  # 1. 加载模型
40
- model_path = "./distilbert-question-detector/checkpoint-5150"
41
  tokenizer = DistilBertTokenizer.from_pretrained(model_path)
42
  model = DistilBertForSequenceClassification.from_pretrained(model_path)
43
  model.eval()
 
37
  from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
38
 
39
  # 1. 加载模型
40
+ model_path = "./distilbert-question-detector"
41
  tokenizer = DistilBertTokenizer.from_pretrained(model_path)
42
  model = DistilBertForSequenceClassification.from_pretrained(model_path)
43
  model.eval()