Update app.py

app.py

@@ -7,13 +7,13 @@ import torch
 import json
 import spaces
 # Load CSV data
-data = pd.read_csv('RB10kstats.csv')
+data = pd.read_csv('RBD10kstats.csv')
 
 # Convert embedding column from JSON string to numpy array
-data['embeddings'] = data['embeddings'].apply(lambda x: np.array(json.loads(x)))
+data['embedding'] = data['embedding'].apply(lambda x: np.array(json.loads(x)))
 
 # Initialize FAISS index
-dimension = len(data['embeddings'][0])
+dimension = len(data['embedding'][0])
 res = faiss.StandardGpuResources()  # use a single GPU
 
 # Check available GPU devices
@@ -24,7 +24,7 @@ if num_gpus > 0:
 else:
     raise RuntimeError("No GPU devices available.")
 
-gpu_index.add(np.stack(data['embeddings'].values))
+gpu_index.add(np.stack(data['embedding'].values))
 
 # Check if GPU is available
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')