asoria HF staff commited on
Commit
513b021
1 Parent(s): 7105d56
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -1,4 +1,3 @@
1
- import spaces
2
  import requests
3
  import logging
4
  import duckdb
@@ -8,6 +7,9 @@ from bertopic.representation import (
8
  KeyBERTInspired,
9
  TextGeneration,
10
  )
 
 
 
11
  from umap import UMAP
12
  import numpy as np
13
  from torch import cuda, bfloat16
@@ -24,6 +26,8 @@ from sklearn.feature_extraction.text import CountVectorizer
24
  # from cuml.cluster import HDBSCAN
25
  # from cuml.manifold import UMAP
26
  from sentence_transformers import SentenceTransformer
 
 
27
  import gradio as gr
28
 
29
  logging.basicConfig(
 
 
1
  import requests
2
  import logging
3
  import duckdb
 
7
  KeyBERTInspired,
8
  TextGeneration,
9
  )
10
+ from bitsandbytes.cuda_setup.main import CUDASetup
11
+ setup = CUDASetup.get_instance()
12
+
13
  from umap import UMAP
14
  import numpy as np
15
  from torch import cuda, bfloat16
 
26
  # from cuml.cluster import HDBSCAN
27
  # from cuml.manifold import UMAP
28
  from sentence_transformers import SentenceTransformer
29
+
30
+ import spaces
31
  import gradio as gr
32
 
33
  logging.basicConfig(