Update app.py
Browse files
app.py
CHANGED
@@ -321,7 +321,7 @@ def strip_title(title):
|
|
321 |
return title
|
322 |
|
323 |
|
324 |
-
def retrieved_info(query, rag_model = rag_model
|
325 |
# Tokenize Query
|
326 |
retriever_input_ids = rag_model.retriever.question_encoder_tokenizer.batch_encode_plus(
|
327 |
[query],
|
|
|
321 |
return title
|
322 |
|
323 |
|
324 |
+
def retrieved_info(query, rag_model = rag_model):
|
325 |
# Tokenize Query
|
326 |
retriever_input_ids = rag_model.retriever.question_encoder_tokenizer.batch_encode_plus(
|
327 |
[query],
|