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