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