Update rag.py

rag.py

@@ -25,8 +25,8 @@ with open('dataset.json', 'r') as f:
     dataset = json.load(f)
 
 # Precompute embeddings
-dataset_questions = [item.get("input", "").lower().strip() for item in dataset]
-dataset_answers = [item.get("response", "") for item in dataset]
+dataset_questions = [item.get("Question", "").lower().strip() for item in dataset]
+dataset_answers = [item.get("Answer", "") for item in dataset]
 dataset_embeddings = similarity_model.encode(dataset_questions, convert_to_tensor=True)
 
 # Greeting list