adding argument to lm.predict()

app.py

@@ -118,7 +118,7 @@ def ner(text):
 
 
   #################### IDSF #######################
-  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id)
+  prediction = lm.predict([fluency_sentence.strip().split()], pred_config, args, tokenizer, pad_token_label_id, device)
   words, slot_preds, intent_pred = prediction[0][0],  prediction[1][0],  prediction[2][0]
 
   slot_tokens = []