vives commited on
Commit
b7a045b
1 Parent(s): 4d020be

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -6,6 +6,7 @@ import torch
6
  import pickle
7
  import numpy as np
8
  import itertools
 
9
 
10
  @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, AutoModelForMaskedLM:lambda _: None})
11
  def load_bert():
 
6
  import pickle
7
  import numpy as np
8
  import itertools
9
+ import tokenizers
10
 
11
  @st.cache(hash_funcs={tokenizers.Tokenizer: lambda _: None, AutoModelForMaskedLM:lambda _: None})
12
  def load_bert():