update

model/model_manager.py

@@ -21,14 +21,10 @@ class ModelManager:
             pipe = self.loaded_models[model_name]
         return pipe
     
-    @spaces.GPU(duration=60)
-    def _generate_image_ig(self, pipe, prompt, model_name):
-        result = pipe(prompt=prompt)
-        return result
-
+    @spaces.GPU(duration=120)
     def generate_image_ig(self, prompt, model_name):
         pipe = self.load_model_pipe(model_name)
-        result = self._generate_image_ig(pipe, prompt, model_name)
+        result = pipe(prompt=prompt)
         return result
 
     def generate_image_ig_parallel_anony(self, prompt, model_A, model_B):
@@ -56,13 +52,9 @@ class ModelManager:
         return results[0], results[1]
 
     @spaces.GPU(duration=150)
-    def _generate_image_ie(self, pipe, source_image, textbox_source, textbox_target, textbox_instruct):
-        result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
-        return result
-    
     def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
         pipe = self.load_model_pipe(model_name)
-        result = self._generate_image_ie(pipe, source_image, textbox_source, textbox_target, textbox_instruct)
+        result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
         return result
 
     def generate_image_ie_parallel(self, textbox_source, textbox_target, textbox_instruct, source_image, model_A, model_B):