Update app.py

app.py

@@ -5,6 +5,7 @@ from llama_index.core import StorageContext, load_index_from_storage, VectorStor
 from llama_index.llms.huggingface import HuggingFaceInferenceAPI
 from llama_index.embeddings.huggingface import HuggingFaceEmbedding
 from sentence_transformers import SentenceTransformer
+import spaces
 
 load_dotenv()
 
@@ -32,6 +33,7 @@ os.makedirs(PERSIST_DIR, exist_ok=True)
 # Variable to store current chat conversation
 current_chat_history = []
 
+@spaces.gpu
 def data_ingestion_from_directory():
     # Use SimpleDirectoryReader on the directory containing the PDF files
     documents = SimpleDirectoryReader(PDF_DIRECTORY).load_data()