hanzla commited on
Commit
9c18d13
·
1 Parent(s): 1fc2033
Files changed (2) hide show
  1. src/app.py +1 -2
  2. src/pdfchatbot.py +0 -3
src/app.py CHANGED
@@ -6,8 +6,7 @@ demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider1 = create
6
 
7
  # Create PDFChatBot instance
8
  pdf_chatbot = PDFChatBot()
9
- pdf_chatbot.load_model()
10
- pdf_chatbot.load_tokenizer()
11
 
12
  # Set up event handlers
13
  with demo:
 
6
 
7
  # Create PDFChatBot instance
8
  pdf_chatbot = PDFChatBot()
9
+ pdf_chatbot.create_organic_pipeline()
 
10
 
11
  # Set up event handlers
12
  with demo:
src/pdfchatbot.py CHANGED
@@ -198,9 +198,6 @@ class PDFChatBot:
198
  self.documents = PyPDFLoader(file.name).load()
199
  self.load_embeddings()
200
  self.load_vectordb()
201
- self.load_tokenizer()
202
- self.load_model()
203
- self.create_organic_pipeline()
204
  #self.create_chain()
205
  @spaces.GPU
206
  def generate_response(self, history, query, file):
 
198
  self.documents = PyPDFLoader(file.name).load()
199
  self.load_embeddings()
200
  self.load_vectordb()
 
 
 
201
  #self.create_chain()
202
  @spaces.GPU
203
  def generate_response(self, history, query, file):