romybeaute commited on
Commit
2e3971c
·
verified ·
1 Parent(s): 322a0bb

doc_info = tm.get_document_info()[["Document", "Topic"]] ## Use the documents stored inside the BERTopic model # to avoid length mismatches if `docs` changes after the run

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -24,6 +24,7 @@ import json
24
  # from huggingface_hub import hf_hub_download, InferenceClient # for the LLM API command
25
  from huggingface_hub import InferenceClient # for the LLM API command
26
 
 
27
 
28
 
29
 
@@ -1173,7 +1174,9 @@ else:
1173
  llm_names = {**default_map, **api_map}
1174
 
1175
 
1176
- doc_info = tm.get_document_info(docs)[["Document", "Topic"]]
 
 
1177
 
1178
  include_outliers = st.checkbox(
1179
  "Include outlier topic (-1)", value=False
 
24
  # from huggingface_hub import hf_hub_download, InferenceClient # for the LLM API command
25
  from huggingface_hub import InferenceClient # for the LLM API command
26
 
27
+ from typing import Any
28
 
29
 
30
 
 
1174
  llm_names = {**default_map, **api_map}
1175
 
1176
 
1177
+ # doc_info = tm.get_document_info(docs)[["Document", "Topic"]]
1178
+ doc_info = tm.get_document_info()[["Document", "Topic"]] ## Use the documents stored inside the BERTopic model
1179
+ # to avoid length mismatches if `docs` changes after the run
1180
 
1181
  include_outliers = st.checkbox(
1182
  "Include outlier topic (-1)", value=False