evanperez commited on
Commit
5a951df
1 Parent(s): 142f4f4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -9,6 +9,7 @@ from langchain.chains.question_answering import load_qa_chain
9
  from langchain.prompts import PromptTemplate
10
  import os
11
  import json
 
12
 
13
  ####CREDIT#####
14
  #Credit to author (Sri Laxmi) of original code reference: SriLaxmi1993
 
9
  from langchain.prompts import PromptTemplate
10
  import os
11
  import json
12
+ from transformers import RagTokenizer, RagRetriever, RagTokenForGeneration
13
 
14
  ####CREDIT#####
15
  #Credit to author (Sri Laxmi) of original code reference: SriLaxmi1993