Spaces:
Runtime error
Runtime error
testing with llama3
Browse files- src/app.py +1 -1
- src/pdfchatbot.py +2 -2
src/app.py
CHANGED
@@ -5,7 +5,7 @@ from pdfchatbot import PDFChatBot
|
|
5 |
demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider1 = create_demo()
|
6 |
|
7 |
# Create PDFChatBot instance
|
8 |
-
pdf_chatbot = PDFChatBot(
|
9 |
pdf_chatbot.load_model()
|
10 |
pdf_chatbot.load_tokenizer()
|
11 |
|
|
|
5 |
demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider1 = create_demo()
|
6 |
|
7 |
# Create PDFChatBot instance
|
8 |
+
pdf_chatbot = PDFChatBot()
|
9 |
pdf_chatbot.load_model()
|
10 |
pdf_chatbot.load_tokenizer()
|
11 |
|
src/pdfchatbot.py
CHANGED
@@ -16,7 +16,7 @@ from langchain_text_splitters import CharacterTextSplitter
|
|
16 |
|
17 |
|
18 |
class PDFChatBot:
|
19 |
-
def __init__(self, config_path="config.yaml"
|
20 |
"""
|
21 |
Initialize the PDFChatBot instance.
|
22 |
|
@@ -37,7 +37,7 @@ class PDFChatBot:
|
|
37 |
self.pipeline = None
|
38 |
self.chain = None
|
39 |
self.chunk_size = None
|
40 |
-
self.chunk_size_slider = chunk_size_slider
|
41 |
|
42 |
def load_config(self, file_path):
|
43 |
"""
|
|
|
16 |
|
17 |
|
18 |
class PDFChatBot:
|
19 |
+
def __init__(self, config_path="config.yaml"):
|
20 |
"""
|
21 |
Initialize the PDFChatBot instance.
|
22 |
|
|
|
37 |
self.pipeline = None
|
38 |
self.chain = None
|
39 |
self.chunk_size = None
|
40 |
+
#self.chunk_size_slider = chunk_size_slider
|
41 |
|
42 |
def load_config(self, file_path):
|
43 |
"""
|