Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -12,7 +12,7 @@ import sys
|
|
| 12 |
import os
|
| 13 |
import gradio as gr
|
| 14 |
|
| 15 |
-
model = load_model('ISCO-Coder-
|
| 16 |
|
| 17 |
bert_layer = hub.KerasLayer("https://kaggle.com/models/tensorflow/bert/TensorFlow2/en-uncased-l-12-h-768-a-12/1",trainable=True)
|
| 18 |
|
|
|
|
| 12 |
import os
|
| 13 |
import gradio as gr
|
| 14 |
|
| 15 |
+
model = load_model('ISCO-Coder-BERT.h5', custom_objects={'KerasLayer': hub.KerasLayer})
|
| 16 |
|
| 17 |
bert_layer = hub.KerasLayer("https://kaggle.com/models/tensorflow/bert/TensorFlow2/en-uncased-l-12-h-768-a-12/1",trainable=True)
|
| 18 |
|