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 |
"""
|