Spaces:
Sleeping
Sleeping
Update inference.py
Browse files- 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,
|
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 |
|