Update inferencer.py

inferencer.py

@@ -121,7 +121,7 @@ class Inferencer:
         model.eval()
         return model
 
-    @spaces.GPU 
+    @spaces.GPU(duration=120)
     def gen_image(
         self,
         raw_prompt,
@@ -166,7 +166,7 @@ class Inferencer:
 
         return [Image.fromarray(img) for img in images[0]]
 
-    @spaces.GPU 
+    @spaces.GPU(duration=120)
     def query_image(self, img: Image.Image, prompt=""):
         model = self.model
         if not model:
@@ -217,7 +217,7 @@ class Inferencer:
         # print(tokenizer.decode(output[0]))
         return tokenizer.decode(output[0])
 
-    @spaces.GPU 
+    @spaces.GPU(duration=120)
     def edit_image(
         self, 
         source_image: Image.Image, 
@@ -331,7 +331,7 @@ class Inferencer:
 
         return [ Image.fromarray(out) ]
     
-    @spaces.GPU 
+    @spaces.GPU(duration=120)
     def query_text(self, prompt=""):
         model = self.model
         if not model: