Update app.py

app.py

@@ -69,7 +69,7 @@ dataset = load_dataset("Amod/mental_health_counseling_conversations")["train"]
 def embed_dataset(_dataset):
     embeddings = []
     for entry in _dataset:
-        embedding = retrieve_embedding(entry["response"])
+        embedding = retrieve_embedding(entry["Response"])
         if embedding is not None:
             embeddings.append(embedding)
     return embeddings
@@ -88,7 +88,7 @@ def retrieve_response(user_query, dataset, dataset_embeddings, k=5):
 
     retrieved_responses = []
     for idx in top_indices:
-        retrieved_responses.append(dataset[idx]["response"])
+        retrieved_responses.append(dataset[idx]["Response"])
     return retrieved_responses
 
 # Streamlit app UI