udpate

model/model_manager.py

@@ -22,9 +22,13 @@ class ModelManager:
         return pipe
     
     @spaces.GPU(duration=60)
+    def _generate_image_ig(self, pipe, prompt, model_name):
+        result = pipe(prompt=prompt)
+        return result
+
     def generate_image_ig(self, prompt, model_name):
         pipe = self.load_model_pipe(model_name)
-        result = pipe(prompt=prompt)
+        result = self._generate_image_ig(pipe, prompt, model_name)
         return result
 
     def generate_image_ig_parallel_anony(self, prompt, model_A, model_B):
@@ -52,9 +56,13 @@ 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 = pipe(src_image = source_image, src_prompt = textbox_source, target_prompt = textbox_target, instruct_prompt = textbox_instruct)
+        result = pipe(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):

serve/utils.py

@@ -128,7 +128,7 @@ def clear_history_side_by_side_ie_anony():
 def get_ip(request: gr.Request):
     if request:
         if "cf-connecting-ip" in request.headers:
-            ip = request.headers["cf-connecting-ip"]
+            ip = request.headers["cf-connecting-ip"] or request.client.host
         else:
             ip = request.client.host
     else: