hanzla commited on
Commit
b6e4be6
·
1 Parent(s): e4b76cd

testing with llama3

Browse files
Files changed (2) hide show
  1. src/app.py +1 -1
  2. 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(chunk_size_slider=slider1)
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",chunk_size_slider=None):
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
  """