Hugging Face's logo

Spaces:
TIGER-Lab
/
GenAI-Arena
Running on Zero

App Files Community
16
DongfuJiang commited on
Commit
ad17295
•
1 Parent(s): 2b3c81e

update

Browse files
Files changed (1) hide show
  1. model/model_manager.py +1 -1
model/model_manager.py CHANGED
@@ -62,7 +62,7 @@ class ModelManager:
62
 
63
  def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
64
  pipe = self.load_model_pipe(model_name)
65
- result = pipe(pipe, src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
66
  return result
67
 
68
  def generate_image_ie_parallel(self, textbox_source, textbox_target, textbox_instruct, source_image, model_A, model_B):
 
62
 
63
  def generate_image_ie(self, textbox_source, textbox_target, textbox_instruct, source_image, model_name):
64
  pipe = self.load_model_pipe(model_name)
65
+ result = self._generate_image_ie(pipe, source_image, textbox_source, textbox_target, textbox_instruct)
66
  return result
67
 
68
  def generate_image_ie_parallel(self, textbox_source, textbox_target, textbox_instruct, source_image, model_A, model_B):