adding arguments to predict()

app.py

@@ -73,6 +73,7 @@ parser.add_argument("--model_dir", default="/home/user/app/Customized_IDSF/Joint
 parser.add_argument("--batch_size", default=32, type=int, help="Batch size for prediction")
 parser.add_argument("--no_cuda", action="store_true", help="Avoid using CUDA when available")
 
+globals()['var'] = "an object"
 pred_config = parser.parse_args()    
 
 # load model and args
@@ -87,11 +88,10 @@ slot_label_lst = get_slot_labels(args)
 pad_token_label_id = args.ignore_index
 tokenizer = load_tokenizer(args) 
 
+#################### END IDSF #######################
 
 
 
-#################### END IDSF #######################
-
 def remove_disfluency(example, prediction):
   characters = list(example)
 
@@ -118,7 +118,7 @@ def ner(text):
 
 
   #################### IDSF #######################
-  prediction = lm.predict([fluency_sentence.strip().split()])
+  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id)
   words, slot_preds, intent_pred = prediction[0][0],  prediction[1][0],  prediction[2][0]
 
   slot_tokens = []