testing with llama3

src/app.py

@@ -5,7 +5,7 @@ from pdfchatbot import PDFChatBot
 demo, chat_history, show_img, txt, submit_button, uploaded_pdf, slider1 = create_demo()
 
 # Create PDFChatBot instance
-pdf_chatbot = PDFChatBot()
+pdf_chatbot = PDFChatBot(chunk_size_slider=slider1)
 pdf_chatbot.load_model()
 pdf_chatbot.load_tokenizer()
 

src/pdfchatbot.py

@@ -16,7 +16,7 @@ from langchain_text_splitters import CharacterTextSplitter
 
 
 class PDFChatBot:
-    def __init__(self, config_path="config.yaml"):
+    def __init__(self, config_path="config.yaml",chunk_size_slider=None):
         """
         Initialize the PDFChatBot instance.
 
@@ -37,6 +37,7 @@ class PDFChatBot:
         self.pipeline = None
         self.chain = None
         self.chunk_size = None
+        self.chunk_size_slider = chunk_size_slider
 
     def load_config(self, file_path):
         """
@@ -70,7 +71,7 @@ class PDFChatBot:
         if not text:
             raise gr.Error('Enter text')
         history.append((text, ''))
-        self.chunk_size.interactive = False
+        self.chunk_size_slider.interactive = False
         return history
 
     def create_prompt_template(self):
@@ -193,6 +194,8 @@ class PDFChatBot:
         Returns:
             PIL.Image.Image: The rendered page as an image.
         """
+        if self.chunk_size_slider.interactive == False:
+            self.chunk_size_slider.interactive = True
         print(chunk_size)
         doc = fitz.open(file.name)
         page = doc[self.page]