adding arguments to lmpredict()

app.py

@@ -81,6 +81,9 @@ args = lm.get_args(pred_config)
 device = lm.get_device(pred_config)
 model = lm.load_model(pred_config, args, device)
 
+intent_label_lst = get_intent_labels(args)
+slot_label_lst = get_slot_labels(args)
+
 # Convert input file to TensorDataset
 pad_token_label_id = args.ignore_index
 tokenizer = load_tokenizer(args) 
@@ -115,7 +118,8 @@ def ner(text):
 
 
   #################### IDSF #######################
-  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id, model, device)
+  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id, model, device,
+                         intent_label_lst, slot_label_lst)
   words, slot_preds, intent_pred = prediction[0][0],  prediction[1][0],  prediction[2][0]
 
   slot_tokens = []