Yoxas commited on
Commit
e0306cb
·
verified ·
1 Parent(s): cc1dc13

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ data = dataset["train"]
27
  # Apply the function to ensure embeddings are 2-dimensional
28
  #data = data.map(lambda example: {'Abstract_Embeddings': ensure_2d_embeddings(example['Abstract_Embeddings'])})
29
 
30
- data = data.add_faiss_index("Abstract")
31
 
32
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
33
 
 
27
  # Apply the function to ensure embeddings are 2-dimensional
28
  #data = data.map(lambda example: {'Abstract_Embeddings': ensure_2d_embeddings(example['Abstract_Embeddings'])})
29
 
30
+ data = data.add_faiss_index("embedding")
31
 
32
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
33