dnj0 commited on
Commit
fe2ae52
·
verified ·
1 Parent(s): a3f184c

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +1 -0
src/app.py CHANGED
@@ -32,6 +32,7 @@ def load_model():
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
33
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
34
  nltk.download('punkt')
 
35
  print("Model loaded!")
36
  return tokenizer, model
37
 
 
32
  tokenizer = AutoTokenizer.from_pretrained(model_name)
33
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
34
  nltk.download('punkt')
35
+ nltk.download('punkt_tab')
36
  print("Model loaded!")
37
  return tokenizer, model
38