Update rag.py

rag.py

@@ -19,6 +19,10 @@ groq_client = Groq(api_key=os.getenv("GROQ_API_KEY"))
 # Load models and dataset
 similarity_model = SentenceTransformer('paraphrase-MiniLM-L6-v2')
 
+# Config
+HF_DATASET_REPO = "midrees2806/unmatched_queries"
+HF_TOKEN = os.getenv("HF_TOKEN")
+
 # Load dataset (automatically using the path)
 with open('dataset.json', 'r') as f:
     dataset = json.load(f)