DD0101 commited on
Commit
0d7455a
1 Parent(s): 0537017

change "fluency_sentence" to list of words for input of predict()

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -110,7 +110,8 @@ def ner(text):
110
 
111
 
112
  #################### IDSF #######################
113
- words, slot_preds, intent_pred = lm.predict(fluency_sentence)[0][0], lm.predict(fluency_sentence)[1][0], lm.predict(fluency_sentence)[2][0]
 
114
 
115
  slot_tokens = []
116
 
 
110
 
111
 
112
  #################### IDSF #######################
113
+ prediction = lm.predict([fluency_sentence.strip().split()])
114
+ words, slot_preds, intent_pred = prediction[0][0], prediction[1][0], prediction[2][0]
115
 
116
  slot_tokens = []
117