Update app.py

app.py

@@ -10,6 +10,8 @@ from langchain.chains import LLMChain
 from langchain.memory import ConversationBufferMemory
 from langchain.prompts import PromptTemplate
 from PyPDF2 import PdfReader
+import spaces
+
 
 class ContextAwareResponseGenerator:
     def __init__(self, llm):
@@ -25,7 +27,7 @@ Chat History: {chat_history}
 Choose the most appropriate response structure and generate the response directly, without explicit guidance on which format to use. Your response should be based on the query and context provided."""  
         )
         self.response_chain = LLMChain(llm=self.llm, prompt=self.response_prompt)
-
+    @spaces.GPU
     def generate_response(self, context, query, chat_history=''):
         try:
             # Generate the response content with structure handled by the LLM itself
@@ -43,6 +45,7 @@ Choose the most appropriate response structure and generate the response directl
         return f"I couldn't generate a response for: {query}"
 
 class AdvancedPdfChatbot:
+    @spaces.GPU
     def __init__(self, openai_api_key):
         os.environ["OPENAI_API_KEY"] = openai_api_key
         self.llm = ChatOpenAI(temperature=0.2, model_name='gpt-4o')
@@ -55,7 +58,8 @@ class AdvancedPdfChatbot:
         
         self.db = None
         self.document_context = ""
-
+        
+    @spaces.GPU
     def load_and_process_pdf(self, pdf_path):
         try:
             reader = PdfReader(pdf_path)
@@ -75,7 +79,8 @@ class AdvancedPdfChatbot:
         except Exception as e:
             logging.error(f"PDF processing error: {e}")
             return False
-
+            
+    @spaces.GPU
     def chat(self, query, is_new_question=False):
         if not self.db:
             return "Please upload a PDF first."