Hugging Face's logo

Spaces:
ccm
/
chat-with-publications
Running on Zero

App Files Community
ccm commited on
Commit
72c49ef
·
verified ·
1 Parent(s): 5e710c8

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -10,7 +10,7 @@ model = sentence_transformers.SentenceTransformer('allenai-specter')
10
  full_data = datasets.load_dataset("ccm/publications")['train'].to_pandas()
11
 
12
  substring = "Designing a New AI"
13
- filter = df['bibtex'].str.contains(substring)
14
  data = full_data[~filter]
15
 
16
  dimensionality = len(data['embedding'][0])
 
10
  full_data = datasets.load_dataset("ccm/publications")['train'].to_pandas()
11
 
12
  substring = "Designing a New AI"
13
+ filter = full_data['bibtex'].str.contains(substring)
14
  data = full_data[~filter]
15
 
16
  dimensionality = len(data['embedding'][0])