Update RagImplementation.py

RagImplementation.py

@@ -13,9 +13,13 @@ from dotenv import load_dotenv
 
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 
+model_name = "Sathvika-Alla/TAL-RAGFallback"
+# tokenizer = AutoTokenizer.from_pretrained(model_name)
+# model = AutoModel.from_pretrained(model_name)
+
 # Load the model and tokenizer
-llm_model = GPT2LMHeadModel.from_pretrained("./results")
-llm_tokenizer = GPT2Tokenizer.from_pretrained("./results")
+llm_model = GPT2LMHeadModel.from_pretrained(model_name)
+llm_tokenizer = GPT2Tokenizer.from_pretrained(model_name)
 llm_tokenizer.pad_token = llm_tokenizer.eos_token