Pradeep Kumar commited on
Commit
a0332ee
·
verified ·
1 Parent(s): 7796c6e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -29,8 +29,10 @@ import gradio as gr
29
 
30
  model = load_model('ISCO-Coder-BERT.h5', custom_objects={'KerasLayer': hub.KerasLayer})
31
 
32
- vocab_file = model.resolved_object.vocab_file.asset_path.numpy()
33
- do_lower_case = model.resolved_object.do_lower_case.numpy()
 
 
34
  tokenizer = tokenization.FullTokenizer(vocab_file,do_lower_case)
35
 
36
  # Parameters
 
29
 
30
  model = load_model('ISCO-Coder-BERT.h5', custom_objects={'KerasLayer': hub.KerasLayer})
31
 
32
+ bert_layer = hub.KerasLayer("https://kaggle.com/models/tensorflow/bert/TensorFlow2/en-uncased-l-12-h-768-a-12/1",trainable=True)
33
+
34
+ vocab_file = bert_layer.resolved_object.vocab_file.asset_path.numpy()
35
+ do_lower_case = bert_layer.resolved_object.do_lower_case.numpy()
36
  tokenizer = tokenization.FullTokenizer(vocab_file,do_lower_case)
37
 
38
  # Parameters