Pradeep Kumar commited on
Commit
abe948e
·
verified ·
1 Parent(s): aecd25a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -1
app.py CHANGED
@@ -38,7 +38,12 @@ tokenizer = tokenization.FullTokenizer(vocab_file,do_lower_case)
38
  # Parameters
39
  max_seq_length = 128
40
  dummy_label = 100
41
- label_list = [i for i in range(1, 425)]
 
 
 
 
 
42
 
43
 
44
  # Define a function to preprocess the new data
 
38
  # Parameters
39
  max_seq_length = 128
40
  dummy_label = 100
41
+
42
+ label_encoder = joblib.load('label_encoder.joblib')
43
+ # Extract the list of labels
44
+ label_classes = label_encoder.classes_
45
+ encoded_values = list(range(len(label_classes)))
46
+ label_list = list(range(len(label_classes)))
47
 
48
 
49
  # Define a function to preprocess the new data