asoria HF staff commited on
Commit
cefd61d
1 Parent(s): 67ba565

Disable cuml again

Browse files
Files changed (2) hide show
  1. app.py +5 -5
  2. 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