new code
Browse files- src/app.py +1 -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.
|
| 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):
|