We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7724551 commit 144ab7bCopy full SHA for 144ab7b
bertopic/representation/_litellm.py
@@ -1,6 +1,7 @@
1
import time
2
from litellm import completion
3
import pandas as pd
4
+from tqdm import tqdm
5
from scipy.sparse import csr_matrix
6
from typing import Mapping, List, Tuple, Any
7
from bertopic.representation._base import BaseRepresentation
@@ -120,7 +121,7 @@ def extract_topics(
120
121
122
# Generate using a (Large) Language Model
123
updated_topics = {}
- for topic, docs in repr_docs_mappings.items():
124
+ for topic, docs in tqdm(repr_docs_mappings.items(), disable=not topic_model.verbose):
125
prompt = self._create_prompt(docs, topic, topics)
126
127
# Delay
0 commit comments