Spaces:
Runtime error
Runtime error
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
|