Update src/pdfchatbot.py

src/pdfchatbot.py

@@ -10,6 +10,8 @@ from langchain.chains import ConversationalRetrievalChain
 from langchain.document_loaders import PyPDFLoader
 from langchain.prompts import PromptTemplate
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
+import spaces
+
 
 class PDFChatBot:
     def __init__(self, config_path="config.yaml"):
@@ -147,7 +149,7 @@ class PDFChatBot:
         self.load_model()
         self.create_pipeline()
         self.create_chain()
-
+    @spaces.GPU
     def generate_response(self, history, query, file):
         """
         Generate a response based on user query and chat history.