Merge branch 'main' of https://huggingface.co/spaces/TIGER-Lab/GenAI-Arena

model/model_manager.py

@@ -54,7 +54,12 @@ class ModelManager:
     @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)
+        if 'PNP' in model_name:
+            result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inversion_steps=100)
+        elif 'Prompt2prompt' in model_name:
+            result = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct, num_inner_steps=5)
+        else:
+            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):

requirements.txt

@@ -46,5 +46,5 @@ openai
 nltk~=3.8.1
 krippendorff
 statsmodels
--e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub
-plotly
+plotly
+-e git+https://github.com/TIGER-AI-Lab/ImagenHub.git#egg=imagen-hub