Update app.py

app.py

@@ -9,14 +9,14 @@ import spaces
 data = pd.read_csv('RB10kstats.csv')
 
 # Convert embedding column from string to numpy array
-data['embedding'] = data['embedding'].apply(lambda x: np.fromstring(x[1:-1], sep=', '))
+data['embeddings'] = data['embeddings'].apply(lambda x: np.fromstring(x[1:-1], sep=', '))
 
 # Initialize FAISS index
-dimension = len(data['embedding'][0])
+dimension = len(data['embeddings'][0])
 res = faiss.StandardGpuResources()  # use a single GPU
 index = faiss.IndexFlatL2(dimension)
 gpu_index = faiss.index_cpu_to_gpu(res, 0, index)  # move to GPU
-gpu_index.add(np.stack(data['embedding'].values))
+gpu_index.add(np.stack(data['embeddings'].values))
 
 # Check if GPU is available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')