not-lain commited on
Commit
b0b771c
β€’
1 Parent(s): 5ea07e3

🌘wπŸŒ–

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -28,7 +28,7 @@ TOP_K = 3
28
  data = load_dataset("not-lain/wikipedia-small-3000-embedded", split="train")
29
 
30
  # index dataset
31
- data.add_faiss_index("embedding", device=1)
32
 
33
 
34
  @spaces.GPU
 
28
  data = load_dataset("not-lain/wikipedia-small-3000-embedded", split="train")
29
 
30
  # index dataset
31
+ data.add_faiss_index("embedding")
32
 
33
 
34
  @spaces.GPU