Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
| 33 |
-
|
|
|
|
|
|
|
| 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
|