Update app.py

app.py

@@ -11,6 +11,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
 from langchain_community.vectorstores import FAISS
 from langchain.schema import Document
 from langchain.embeddings import HuggingFaceEmbeddings
+import spaces  # Add this import for Hugging Face Spaces
 
 # Create the Vision 2030 Assistant class
 class Vision2030Assistant:
@@ -86,6 +87,7 @@ def retrieve_context(query, vector_store, top_k=5):
     
     return contexts
 
+@spaces.GPU  # Add decorator for GPU usage
 def generate_response(query, contexts, model, tokenizer, language="auto"):
     """Generate a response using retrieved contexts with ALLaM-specific formatting"""
     # Auto-detect language if not specified
@@ -221,6 +223,7 @@ tokenizer = None
 assistant = None
 
 # Load the model and tokenizer
+@spaces.GPU  # Add decorator for GPU usage
 def load_model_and_tokenizer():
     global model, tokenizer
     
@@ -294,6 +297,7 @@ def process_pdfs(pdf_files):
     
     return f"Successfully processed {len(documents)} documents. The assistant is ready to use!"
 
+@spaces.GPU  # Add decorator for GPU usage
 def answer_query(message, history):
     global assistant