Update app.py

app.py

@@ -81,9 +81,6 @@ 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) 
@@ -118,7 +115,7 @@ def ner(text):
 
 
   #################### IDSF #######################
-  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id, device)
+  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id, model, device)
   words, slot_preds, intent_pred = prediction[0][0],  prediction[1][0],  prediction[2][0]
 
   slot_tokens = []