Update app.py

app.py

@@ -30,8 +30,8 @@ class DocumentRetrievalAndGeneration:
         all_splits = text_splitter.split_documents(documents)
         print('Length of documents:', len(documents))
         print("LEN of all_splits", len(all_splits))
-        for i in range(5):
-            print(all_splits[i].page_content)
+        # for i in range(5):
+        #     print(all_splits[i].page_content)
         return all_splits
 
     def create_faiss_index(self):
@@ -67,7 +67,7 @@ class DocumentRetrievalAndGeneration:
         try:
             with concurrent.futures.ThreadPoolExecutor() as executor:
                 future = executor.submit(self.llm.model.generate, model_inputs, max_new_tokens=1000, do_sample=True)
-                generated_ids = future.result(timeout=60)  # Timeout set to 60 seconds
+                generated_ids = future.result(timeout=80)  # Timeout set to 60 seconds
             return generated_ids
         except concurrent.futures.TimeoutError:
             return "Text generation process timed out"