Update app.py

app.py

@@ -69,15 +69,18 @@ class CFG:
     PDFs_path = './data'
     Embeddings_path =  './embeddings/input'
     Output_folder = './ml-papers-vector'
-
+    
+@spaces.GPU(duration=120)
 loader = DirectoryLoader(CFG.PDFs_path, glob="./*.pdf", loader_cls=PyPDFLoader,use_multithreading=True)
 
+@spaces.GPU(duration=120)
 documents = loader.load()
 
+@spaces.GPU(duration=120)
 text_splitter = RecursiveCharacterTextSplitter(chunk_size = CFG.split_chunk_size, chunk_overlap = CFG.split_overlap)
-
+@spaces.GPU(duration=120)
 texts = text_splitter.split_documents(documents)
-
+@spaces.GPU(duration=120)
 if not os.path.exists(CFG.Embeddings_path + '/index.faiss'):
     embeddings = HuggingFaceInstructEmbeddings(model_name = CFG.embeddings_model_repo, model_kwargs={"device":"cuda"})
     vectordb = FAISS.from_documents(documents=texts, embedding=embeddings)