Spaces:

NLPV
/

ISCO-code-predictor-api

Sleeping

NLPV commited on Aug 20, 2024

Commit

c14dafe

verified ·

1 Parent(s): d8f0fd4

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import sys
 import os
 import gradio as gr
-model = load_model('ISCO-Coder-BERT.h5', custom_objects={'KerasLayer': hub.KerasLayer})
 bert_layer = hub.KerasLayer("https://kaggle.com/models/tensorflow/bert/TensorFlow2/en-uncased-l-12-h-768-a-12/1",trainable=True)

 import os
 import gradio as gr
+model = load_model('ISCO-Coder-BERTV2.h5', custom_objects={'KerasLayer': hub.KerasLayer})
 bert_layer = hub.KerasLayer("https://kaggle.com/models/tensorflow/bert/TensorFlow2/en-uncased-l-12-h-768-a-12/1",trainable=True)