Update inferencer.py

inferencer.py

@@ -122,6 +122,7 @@ class Inferencer:
         model.eval()
         return model
 
+    @spaces.GPU 
     def gen_image(
         self,
         raw_prompt,
@@ -164,6 +165,7 @@ class Inferencer:
 
         return [Image.fromarray(img) for img in images[0]]
 
+    @spaces.GPU 
     def query_image(self, img: Image.Image, prompt=""):
         model = self.model
         tokenizer = model.tokenizer
@@ -349,6 +351,7 @@ class Inferencer:
     #     )
     #     return [output_image]
 
+    @spaces.GPU 
     def edit_image(
         self, 
         source_image: Image.Image, 
@@ -460,6 +463,7 @@ class Inferencer:
 
         return [ Image.fromarray(out) ]
     
+    @spaces.GPU 
     def query_text(self, prompt=""):
         model = self.model
         tokenizer = model.tokenizer