Update modules/pmbl.py

modules/pmbl.py

@@ -152,7 +152,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, n_gpu_layers=-1, mlock=True)
         llm = llm.to("cuda")  # Move the model to the GPU