EnverLee commited on
Commit
e1facd9
1 Parent(s): 9d802ee

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ ST = SentenceTransformer("mixedbread-ai/mxbai-embed-large-v1")
16
  dataset = load_dataset("jihye-moon/LawQA-Ko")
17
 
18
  data = dataset["train"]
19
- data = data.add_faiss_index("embeddings") # column name that has the embeddings of the dataset
20
 
21
 
22
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
 
16
  dataset = load_dataset("jihye-moon/LawQA-Ko")
17
 
18
  data = dataset["train"]
19
+ data = data.add_faiss_index("question", "answer") # column name that has the embeddings of the dataset
20
 
21
 
22
  model_id = "meta-llama/Meta-Llama-3-8B-Instruct"