arif670 commited on
Commit
03f2920
·
verified ·
1 Parent(s): ceaf791

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ firebase_admin.initialize_app(cred, {"databaseURL": "https://your-database-name.
20
 
21
  # Load the RAG model, tokenizer, and retriever
22
  tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-base")
23
- retriever = RagRetriever.from_pretrained("facebook/rag-token-base", use_dummy_dataset=True) # Use a dummy dataset for now
24
  model = RagSequenceForGeneration.from_pretrained("facebook/rag-token-base")
25
 
26
  # Function to generate answers using the RAG model
 
20
 
21
  # Load the RAG model, tokenizer, and retriever
22
  tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-base")
23
+ retriever = RagRetriever.from_pretrained("facebook/rag-token-base", use_dummy_dataset=True, trust_remote_code=True)
24
  model = RagSequenceForGeneration.from_pretrained("facebook/rag-token-base")
25
 
26
  # Function to generate answers using the RAG model