jaimin commited on
Commit
dc45be0
·
1 Parent(s): 771d137

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -15,6 +15,8 @@ import re
15
  import torch
16
  from transformers import BertTokenizer, BertModel, BertForMaskedLM
17
  from wordfreq import zipf_frequency
 
 
18
 
19
  bert_model = 'bert-large-uncased'
20
  tokenizer = BertTokenizer.from_pretrained(bert_model)
 
15
  import torch
16
  from transformers import BertTokenizer, BertModel, BertForMaskedLM
17
  from wordfreq import zipf_frequency
18
+ from keras.preprocessing.sequence import pad_sequences
19
+ from tensorflow.keras.utils import to_categorical
20
 
21
  bert_model = 'bert-large-uncased'
22
  tokenizer = BertTokenizer.from_pretrained(bert_model)