Kalyani8 commited on
Commit
545f3e4
·
verified ·
1 Parent(s): 7b7b4f7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,8 +4,8 @@ import faiss
4
  import numpy as np
5
 
6
 
7
- # Load a small subset (12,000 rows)
8
- dataset = load_dataset("wiki40b", "en", split="train[:12000]")
9
 
10
  # Extract only text
11
  docs = [d["text"] for d in dataset]
 
4
  import numpy as np
5
 
6
 
7
+ # Load a small subset (10,000 rows)
8
+ dataset = load_dataset("wiki40b", "en", split="train[:10000]")
9
 
10
  # Extract only text
11
  docs = [d["text"] for d in dataset]