NLPV commited on
Commit
7e3c140
·
verified ·
1 Parent(s): cb2687e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -12,7 +12,7 @@ import sys
12
  import os
13
  import gradio as gr
14
 
15
- model = load_model('ISCO-Coder-BERTV2.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
 
@@ -23,11 +23,11 @@ tokenizer = tokenization.FullTokenizer(vocab_file,do_lower_case)
23
  # Parameters
24
  max_seq_length = 128
25
  dummy_label = 100
26
- label_list = list(pd.read_excel('label_listV2.xlsx')['label_list'])
27
 
28
  map_data = pd.read_excel("ISCO-08 EN Structure and definitions.xlsx")
29
 
30
- label_encoder = joblib.load('label_encoderV2.joblib')
31
 
32
  # Define a function to preprocess the new data
33
  def get_feature_new(text, max_seq_length, tokenizer, dummy_label):
 
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
 
 
23
  # Parameters
24
  max_seq_length = 128
25
  dummy_label = 100
26
+ label_list = list(pd.read_excel('label_list.xlsx')['label_list'])
27
 
28
  map_data = pd.read_excel("ISCO-08 EN Structure and definitions.xlsx")
29
 
30
+ label_encoder = joblib.load('label_encoder.joblib')
31
 
32
  # Define a function to preprocess the new data
33
  def get_feature_new(text, max_seq_length, tokenizer, dummy_label):