HugoHE commited on
Commit
8b34a32
1 Parent(s): 9dacf41

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import pipeline, BertForSequenceClassification, BertTokenizer
4
 
5
  def classify(input_text):
6
  tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
7
- model = BertForSequenceClassification.from_pretrained('.')
8
  classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=3)
9
  class_dict = {0:'story',
10
  1:'culture',
 
4
 
5
  def classify(input_text):
6
  tokenizer = BertTokenizer.from_pretrained('bert-base-chinese')
7
+ model = BertForSequenceClassification.from_pretrained('./bert-cls')
8
  classifier = pipeline("text-classification", model=model, tokenizer=tokenizer, top_k=3)
9
  class_dict = {0:'story',
10
  1:'culture',