Update src/pdfchatbot.py

src/pdfchatbot.py

@@ -57,11 +57,11 @@ class PDFChatBot:
         docs = text_splitter.split_documents(self.documents)
         self.vectordb = Chroma.from_documents(docs, self.embeddings)
         print("Vector store created")
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     def load_tokenizer(self):
         self.tokenizer = AutoTokenizer.from_pretrained("gradientai/Llama-3-8B-Instruct-Gradient-1048k")
 
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     def create_organic_pipeline(self):
         self.pipe = pipeline(
             "text-generation",
@@ -77,7 +77,7 @@ class PDFChatBot:
         self.current_context = context
         print("Context Ready")
         print(self.current_context)
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     def create_organic_response(self, history, query):
         self.get_organic_context(query)
         """
@@ -121,7 +121,7 @@ class PDFChatBot:
         self.load_vectordb()
         self.create_organic_pipeline()
         #self.create_chain()
-    @spaces.GPU
+    @spaces.GPU(duration=120)
     def generate_response(self, history, query, file,chunk_size,chunk_overlap_percentage,model_temperature,max_chunks_in_context):
 
         self.chunk_size = chunk_size