removing Thread

app.py

@@ -47,7 +47,7 @@ class BSIChatbot:
         self.docs = docs_path
         self.rerank_model_path = model_paths['rerank_model_path']
 
-    #@spaces.GPU
+    @spaces.GPU
     def initialize_embedding_model(self, rebuild_embeddings: bool):
         raw_knowledge_base = []
 
@@ -94,7 +94,7 @@ class BSIChatbot:
             self.vectorstore = FAISS.load_local(os.path.join(self.docs, "_embeddings"), self.embedding_model, allow_dangerous_deserialization=True)
             print("DBG: Vectorstore Status Initialization:", self.vectorstore)
 
-    #@spaces.GPU
+    @spaces.GPU
     def retrieve_similar_embedding(self, query: str):
         #lazy load
         #if (self.vectorstore == None):
@@ -104,7 +104,7 @@ class BSIChatbot:
         query = f"Instruct: Given a search query, retrieve the relevant passages that answer the query\nQuery:{query}"
         return self.vectorstore.similarity_search(query=query, k=20)
 
-    #@spaces.GPU
+    @spaces.GPU
     def initialize_llm(self):
         bnb_config = BitsAndBytesConfig(load_in_8bit=True)
         llm = AutoModelForCausalLM.from_pretrained(self.llm_path, quantization_config=bnb_config)