Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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,
|