Spaces:
Sleeping
Sleeping
Disable cuml again
Browse files- app.py +5 -5
- requirements.txt +2 -2
app.py
CHANGED
@@ -27,6 +27,11 @@ from transformers import (
|
|
27 |
AutoModelForCausalLM,
|
28 |
pipeline,
|
29 |
)
|
|
|
|
|
|
|
|
|
|
|
30 |
|
31 |
"""
|
32 |
TODOs:
|
@@ -137,9 +142,6 @@ def calculate_n_neighbors_and_components(n_rows):
|
|
137 |
|
138 |
# @spaces.GPU
|
139 |
def fit_model(docs, embeddings, n_neighbors, n_components):
|
140 |
-
from cuml.manifold import UMAP
|
141 |
-
from cuml.cluster import HDBSCAN
|
142 |
-
|
143 |
umap_model = UMAP(
|
144 |
n_neighbors=n_neighbors,
|
145 |
n_components=n_components,
|
@@ -237,8 +239,6 @@ datasets:
|
|
237 |
|
238 |
@spaces.GPU(duration=120)
|
239 |
def generate_topics(dataset, config, split, column, nested_column, plot_type):
|
240 |
-
from cuml.manifold import UMAP
|
241 |
-
|
242 |
logging.info(
|
243 |
f"Generating topics for {dataset} with config {config} {split} {column} {nested_column}"
|
244 |
)
|
|
|
27 |
AutoModelForCausalLM,
|
28 |
pipeline,
|
29 |
)
|
30 |
+
# from cuml.manifold import UMAP
|
31 |
+
# from cuml.cluster import HDBSCAN
|
32 |
+
|
33 |
+
from umap import UMAP
|
34 |
+
from hdbscan import HDBSCAN
|
35 |
|
36 |
"""
|
37 |
TODOs:
|
|
|
142 |
|
143 |
# @spaces.GPU
|
144 |
def fit_model(docs, embeddings, n_neighbors, n_components):
|
|
|
|
|
|
|
145 |
umap_model = UMAP(
|
146 |
n_neighbors=n_neighbors,
|
147 |
n_components=n_components,
|
|
|
239 |
|
240 |
@spaces.GPU(duration=120)
|
241 |
def generate_topics(dataset, config, split, column, nested_column, plot_type):
|
|
|
|
|
242 |
logging.info(
|
243 |
f"Generating topics for {dataset} with config {config} {split} {column} {nested_column}"
|
244 |
)
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
-
--extra-index-url https://pypi.nvidia.com
|
2 |
-
cuml-cu11
|
3 |
spaces
|
4 |
gradio
|
5 |
torch
|
|
|
1 |
+
# --extra-index-url https://pypi.nvidia.com
|
2 |
+
# cuml-cu11
|
3 |
spaces
|
4 |
gradio
|
5 |
torch
|