Update app.py

app.py

@@ -19,19 +19,19 @@ data = dataset["train"]
 # Convert the string embeddings to numerical arrays
 def convert_and_ensure_2d_embeddings(example):
     # Convert the string to a numpy array
-    embedding_str = example['Abstract']
+    embedding_str = example['embedding']
     embedding_str = embedding_str.replace('\n', ' ')
     embedding_list = list(map(float, embedding_str.strip("[]").split()))
     embeddings = np.array(embedding_list, dtype=np.float32)
     # Ensure the embeddings are 2-dimensional
     if embeddings.ndim == 1:
         embeddings = embeddings.reshape(1, -1)
-    return {'Abstract': embeddings}
+    return {'embedding': embeddings}
 
 # Apply the function to ensure embeddings are 2-dimensional and of type float32
 data = data.map(convert_and_ensure_2d_embeddings)
 
-data = data.add_faiss_index("Abstract")
+data = data.add_faiss_index("embedding")
 
 model_id = "meta-llama/Meta-Llama-3-8B-Instruct"