Update app.py

app.py

@@ -9,7 +9,7 @@ from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 import os
 import json
-from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
+from transformers import AutoTokenizer
 
 ####CREDIT#####
 #Credit to author (Sri Laxmi) of original code reference: SriLaxmi1993
@@ -17,10 +17,9 @@ from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
 #Sri Laxmi Youtube:https://www.youtube.com/watch?v=SkY2u4UUr6M&t=112s
 ###############
 os.system("pip install -r requirements.txt")
-#Model loading
-tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")
-retriever = RagRetriever.from_pretrained("facebook/rag-token-nq", index_name="exact", use_dummy_dataset=True)
-model = RagTokenForGeneration.from_pretrained("facebook/rag-token-nq", retriever=retriever)
+
+#Model
+tokenizer = AutoTokenizer.from_pretrained("bofenghuang/vigogne-2-7b-chat")
 
 st.set_page_config(page_title="Gemini RAG", layout="wide")