ogawaal commited on
Commit
d29ce9c
·
verified ·
1 Parent(s): 2a6cc02

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +1 -1
inference.py CHANGED
@@ -219,5 +219,5 @@ def infer_topic(new_text: str) -> tuple[np.ndarray, float]:
219
  new_word_list = tokenizer_sudachi.get_wakati(new_text)
220
  new_doc = model_trained.make_doc(new_word_list)
221
  topic_dist, ll = model_trained.infer(new_doc)
222
- return topic_dist,new_word_list, ll
223
 
 
219
  new_word_list = tokenizer_sudachi.get_wakati(new_text)
220
  new_doc = model_trained.make_doc(new_word_list)
221
  topic_dist, ll = model_trained.infer(new_doc)
222
+ return topic_dist, ll
223