Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
55a0b15
1
Parent(s):
9b3489f
Remove unused dataset_name import and HF_TOKEN parameter from DatasetManager initialization
Browse files
app.py
CHANGED
@@ -9,7 +9,7 @@ from huggingface_hub import InferenceClient
|
|
9 |
from sentence_transformers import SentenceTransformer
|
10 |
|
11 |
from arxiv_stuff import ARXIV_CATEGORIES_FLAT
|
12 |
-
from dataset_utils import DatasetManager
|
13 |
|
14 |
# Get HF_TOKEN from environment variables
|
15 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
@@ -745,8 +745,6 @@ if __name__ == "__main__":
|
|
745 |
|
746 |
# Load dataset with FAISS index
|
747 |
dataset = DatasetManager(
|
748 |
-
dataset_name=dataset_name,
|
749 |
-
hf_token=HF_TOKEN,
|
750 |
embedding_model=embedding_model,
|
751 |
)
|
752 |
|
|
|
9 |
from sentence_transformers import SentenceTransformer
|
10 |
|
11 |
from arxiv_stuff import ARXIV_CATEGORIES_FLAT
|
12 |
+
from dataset_utils import DatasetManager
|
13 |
|
14 |
# Get HF_TOKEN from environment variables
|
15 |
HF_TOKEN = os.getenv("HF_TOKEN")
|
|
|
745 |
|
746 |
# Load dataset with FAISS index
|
747 |
dataset = DatasetManager(
|
|
|
|
|
748 |
embedding_model=embedding_model,
|
749 |
)
|
750 |
|