Moha782 commited on
Commit
d5583bb
·
verified ·
1 Parent(s): e6843a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
9
 
10
  # Load the RAG tokenizer and retriever
11
  tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")
12
- retriever = RagRetriever.from_pretrained("facebook/rag-token-nq", index_name="apexcustoms", passages="apexcustoms.pdf")
13
 
14
  def respond(
15
  message,
 
9
 
10
  # Load the RAG tokenizer and retriever
11
  tokenizer = RagTokenizer.from_pretrained("facebook/rag-token-nq")
12
+ retriever = RagRetriever.from_pretrained("facebook/rag-token-nq", index_name="apexcustoms", passages="apexcustoms.pdf", trust_remote_code=True)
13
 
14
  def respond(
15
  message,