Update src/RAGSample.py

src/RAGSample.py

@@ -17,6 +17,7 @@ from sklearn.metrics.pairwise import cosine_similarity
 import pandas as pd
 from typing import Optional, List
 import re
+import torch
 
 # OPTION 1: Use Hugging Face Pipeline (Recommended for HF Spaces)
 from transformers import pipeline