asoria HF staff commited on
Commit
43d95a6
1 Parent(s): 2269797

Enabling UMAP and HDBSCAN with cuml again

Browse files
Files changed (2) hide show
  1. app.py +4 -4
  2. requirements.txt +2 -0
app.py CHANGED
@@ -25,10 +25,10 @@ from bertopic.representation import KeyBERTInspired
25
  from bertopic.representation import TextGeneration
26
 
27
  # Temporary disabling because of ZeroGPU does not support cuml
28
- # from cuml.manifold import UMAP
29
- # from cuml.cluster import HDBSCAN
30
- from umap import UMAP
31
- from hdbscan import HDBSCAN
32
  from huggingface_hub import HfApi
33
  from sklearn.feature_extraction.text import CountVectorizer
34
  from sentence_transformers import SentenceTransformer
 
25
  from bertopic.representation import TextGeneration
26
 
27
  # Temporary disabling because of ZeroGPU does not support cuml
28
+ from cuml.manifold import UMAP
29
+ from cuml.cluster import HDBSCAN
30
+ # from umap import UMAP
31
+ # from hdbscan import HDBSCAN
32
  from huggingface_hub import HfApi
33
  from sklearn.feature_extraction.text import CountVectorizer
34
  from sentence_transformers import SentenceTransformer
requirements.txt CHANGED
@@ -14,3 +14,5 @@ numpy
14
  python-dotenv
15
  kaleido
16
  transformers
 
 
 
14
  python-dotenv
15
  kaleido
16
  transformers
17
+ --extra-index-url https://pypi.nvidia.com
18
+ cuml-cu11