Update retrieval_qa_pipeline.py

retrieval_qa_pipeline.py

@@ -86,8 +86,8 @@ def build_retrieval_qa_pipeline(model, tokenizer, vectorstore):
 
 def main():
     # Replace these names with your model and dataset repo names
-    model_name = "username/my_fine_tuned_model"
-    dataset_name = "PranavKeshav/upf_code"
+    model_name = "anirudh248/upf_code_generator_final"
+    dataset_name = "PranavKeshav/upf_dataset"
     faiss_index_path = "faiss_index"
     
     print("Starting pipeline setup...")