Spaces:
Sleeping
Sleeping
Try fix
Browse files
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(
|