hHoai commited on
Commit
d016577
·
verified ·
1 Parent(s): 9ec8da8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,6 @@
1
  from model.model import Seq2SeqModel, predict, model_tokenize
 
 
2
 
3
  model_args = {
4
  "max_seq_length": 512,
 
1
  from model.model import Seq2SeqModel, predict, model_tokenize
2
+ from model.model import Seq2SeqModel, predict, model_tokenize
3
+ from transformers import AutoTokenizer
4
 
5
  model_args = {
6
  "max_seq_length": 512,