Huangxin14 commited on
Commit
862a2ae
·
verified ·
1 Parent(s): 2fdb68e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -10,7 +10,6 @@ model_dir = "my-bert-model"
10
 
11
  config = AutoConfig.from_pretrained(
12
  model_dir,
13
- num_labels=3,
14
  finetuning_task="text-classification"
15
  )
16
  tokenizer = AutoTokenizer.from_pretrained(model_dir)
 
10
 
11
  config = AutoConfig.from_pretrained(
12
  model_dir,
 
13
  finetuning_task="text-classification"
14
  )
15
  tokenizer = AutoTokenizer.from_pretrained(model_dir)