thechaiexperiment commited on
Commit
629c5b2
·
1 Parent(s): 9e47f21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -13,7 +13,8 @@ from transformers import (
13
  AutoTokenizer,
14
  BartForConditionalGeneration,
15
  AutoModelForCausalLM,
16
- AutoModelForSeq2SeqLM
 
17
  )
18
  import pandas as pd
19
  import time
 
13
  AutoTokenizer,
14
  BartForConditionalGeneration,
15
  AutoModelForCausalLM,
16
+ AutoModelForSeq2SeqLM,
17
+ AutoModelForTokenClassification
18
  )
19
  import pandas as pd
20
  import time