Pupolina commited on
Commit
f415df8
·
verified ·
1 Parent(s): 0cb73d5

intermediate check

Browse files
Files changed (1) hide show
  1. functionality.py +2 -2
functionality.py CHANGED
@@ -112,9 +112,9 @@ def get_collection() -> chromadb.Collection:
112
  papers = pd.read_csv("hf://datasets/somosnlp-hackathon-2022/scientific_papers_en/scientific_paper_en.csv")
113
  logging.info(f"The data downloaded from url.")
114
  papers = papers.drop(['id'], axis=1)
115
- papers = papers.iloc[:200]
116
 
117
- for i in range(200):
118
  paper = papers.iloc[i]
119
  idx = paper.name
120
 
 
112
  papers = pd.read_csv("hf://datasets/somosnlp-hackathon-2022/scientific_papers_en/scientific_paper_en.csv")
113
  logging.info(f"The data downloaded from url.")
114
  papers = papers.drop(['id'], axis=1)
115
+ papers = papers.iloc[:100]
116
 
117
+ for i in range(100):
118
  paper = papers.iloc[i]
119
  idx = paper.name
120