Update modules/pmbl.py

modules/pmbl.py

@@ -2,7 +2,6 @@ import sqlite3
 from datetime import datetime
 from llama_cpp import Llama
 from concurrent.futures import ThreadPoolExecutor
-import spaces
 
 class PMBL:
     def __init__(self, model_path):
@@ -102,7 +101,7 @@ class PMBL:
         for chunk in response.result():
             yield chunk
 
-    @spaces.GPU
+    
     def generate_response_task(self, system_prompt, prompt, n_ctx):
         llm = Llama(model_path=self.model_path, n_ctx=n_ctx, n_threads=8, mlock=True, n_gpu_layers=42)
 
@@ -149,7 +148,7 @@ class PMBL:
 
         conn.close()
 
-    @spaces.GPU
+    
     def generate_topic(self, prompt, response):
         llm = Llama(model_path=self.model_path, n_ctx=1690, n_threads=8, mlock=True, n_gpu_layers=42)