nomadicsynth commited on
Commit
55a0b15
·
1 Parent(s): 9b3489f

Remove unused dataset_name import and HF_TOKEN parameter from DatasetManager initialization

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