Update app.py

app.py

@@ -9,12 +9,12 @@ import spaces
 data = pd.read_csv('RBDx10stats.csv')
 
 # Convert embedding column from string to numpy array
-data['Embedding'] = data['Embedding'].apply(lambda x: np.fromstring(x[1:-1], sep=', '))
+data['embedding'] = data['embedding'].apply(lambda x: np.fromstring(x[1:-1], sep=', '))
 
 # Initialize FAISS index
-dimension = len(data['Embedding'][0])
+dimension = len(data['embedding'][0])
 index = faiss.IndexFlatL2(dimension)
-index.add(np.stack(data['Embedding'].values))
+index.add(np.stack(data['embedding'].values))
 
 # Load QA model
 qa_model = pipeline("question-answering", model="distilbert-base-uncased-distilled-squad")