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