momo commited on
Commit
faf7c96
โ€ข
1 Parent(s): 2a1fcfd
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -14,10 +14,9 @@ model_name_list = [
14
  ]
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
- model_name,
18
- num_labels=15,
19
- problem_type="multi_label_classification"
20
  )
 
21
  tokenizer = AutoTokenizer.from_pretrained(model_name)
22
 
23
  unsmile_labels = ["์—ฌ์„ฑ/๊ฐ€์กฑ","๋‚จ์„ฑ","์„ฑ์†Œ์ˆ˜์ž","์ธ์ข…/๊ตญ์ ","์—ฐ๋ น","์ง€์—ญ","์ข…๊ต","๊ธฐํƒ€ ํ˜์˜ค","์•…ํ”Œ/์š•์„ค","clean", 'name', 'number', 'address', 'bank', 'person']
 
14
  ]
15
 
16
  model = BertForSequenceClassification.from_pretrained(
17
+ model_name
 
 
18
  )
19
+
20
  tokenizer = AutoTokenizer.from_pretrained(model_name)
21
 
22
  unsmile_labels = ["์—ฌ์„ฑ/๊ฐ€์กฑ","๋‚จ์„ฑ","์„ฑ์†Œ์ˆ˜์ž","์ธ์ข…/๊ตญ์ ","์—ฐ๋ น","์ง€์—ญ","์ข…๊ต","๊ธฐํƒ€ ํ˜์˜ค","์•…ํ”Œ/์š•์„ค","clean", 'name', 'number', 'address', 'bank', 'person']