Baweja commited on
Commit
8294e1c
·
verified ·
1 Parent(s): 579a454

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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, generating_model = 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],