Wikipedia_RAG / app.py
Kalyani8's picture
Update app.py
87188d3 verified
raw
history blame
245 Bytes
from datasets import load_dataset
# Load a small subset (12,000 rows)
dataset = load_dataset("wiki40b", "en", split="train[:12000]")
# Extract only text
docs = [d["text"] for d in dataset]
print("Loaded dataset with", len(docs), "documents.")