update

model/model_manager.py

@@ -3,6 +3,7 @@ import random
 import gradio as gr
 import requests
 import io, base64, json
+import spaces
 from PIL import Image
 from .models import IMAGE_GENERATION_MODELS, IMAGE_EDITION_MODELS, load_pipeline
 
@@ -19,7 +20,8 @@ class ModelManager:
         else:
             pipe = self.loaded_models[model_name]
         return pipe
-
+    
+    @spaces.GPU(duration=60)
     def generate_image_ig(self, prompt, model_name):
         pipe = self.load_model_pipe(model_name)
         result = pipe(prompt=prompt)
@@ -49,6 +51,7 @@ class ModelManager:
                 results.append(result)
         return results[0], results[1]
 
+    @spaces.GPU(duration=150)
     def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
         pipe = self.load_model_pipe(model_name)
         result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)