Spaces:
Sleeping
Sleeping
Enabling UMAP and HDBSCAN with cuml again
Browse files- app.py +4 -4
- 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 |
-
|
29 |
-
|
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
|