Spaces:
Runtime error
Runtime error
i dont remember adding security but there are secutity issues
Browse files
rag_pipeline/data/load_data.py
CHANGED
|
@@ -15,7 +15,7 @@ def clean_text(text):
|
|
| 15 |
|
| 16 |
|
| 17 |
def load_arxiv_subset(n: int = 10000):
|
| 18 |
-
dataset = load_dataset("ccdv/arxiv-summarization")
|
| 19 |
return dataset["train"].select(range(n))
|
| 20 |
|
| 21 |
|
|
@@ -39,7 +39,7 @@ def process_dataset(data) -> list[dict]:
|
|
| 39 |
if len(chunk) < 50:
|
| 40 |
continue
|
| 41 |
all_chunks.append({
|
| 42 |
-
"chunk_id": hashlib.md5(f"{i}_abstract_{chunk_idx}".encode()).hexdigest(),
|
| 43 |
"paper_id": i,
|
| 44 |
"section_title": "ABSTRACT",
|
| 45 |
"chunk_index": chunk_idx,
|
|
@@ -54,7 +54,7 @@ def process_dataset(data) -> list[dict]:
|
|
| 54 |
if len(chunk) < 50:
|
| 55 |
continue
|
| 56 |
all_chunks.append({
|
| 57 |
-
"chunk_id": hashlib.md5(f"{i}_article_{chunk_idx}".encode()).hexdigest(),
|
| 58 |
"paper_id": i,
|
| 59 |
"section_title": "ARTICLE",
|
| 60 |
"chunk_index": chunk_idx,
|
|
|
|
| 15 |
|
| 16 |
|
| 17 |
def load_arxiv_subset(n: int = 10000):
|
| 18 |
+
dataset = load_dataset("ccdv/arxiv-summarization",revision="main")
|
| 19 |
return dataset["train"].select(range(n))
|
| 20 |
|
| 21 |
|
|
|
|
| 39 |
if len(chunk) < 50:
|
| 40 |
continue
|
| 41 |
all_chunks.append({
|
| 42 |
+
"chunk_id": hashlib.md5(f"{i}_abstract_{chunk_idx}".encode(),usedforsecurity=False).hexdigest(),
|
| 43 |
"paper_id": i,
|
| 44 |
"section_title": "ABSTRACT",
|
| 45 |
"chunk_index": chunk_idx,
|
|
|
|
| 54 |
if len(chunk) < 50:
|
| 55 |
continue
|
| 56 |
all_chunks.append({
|
| 57 |
+
"chunk_id": hashlib.md5(f"{i}_article_{chunk_idx}".encode(),usedforsecurity=False).hexdigest(),
|
| 58 |
"paper_id": i,
|
| 59 |
"section_title": "ARTICLE",
|
| 60 |
"chunk_index": chunk_idx,
|