feat: Refactorizar función de refinamiento de imágenes en app.py

app.py

@@ -134,29 +134,11 @@ def save_image(image):
     image.save(unique_filename)
     return unique_filename
 
-def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
-    if (not apply_refiner):
-        return image
-    
-    client = InferenceClient()
-    refined_image = client.image_to_image(
-        handle_file(image), 
-        prompt=prompt, 
-        negative_prompt=negative_prompt, 
-        num_inference_steps=num_inference_steps, 
-        guidance_scale=guidance_scale, 
-        seed=seed, 
-        model=model,
-        strength=strength
-    )
-    return refined_image
-
-
 
 def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name, reduce_factor):
     global client_tile_upscaler
 
-    image = image[1]
+    # image = image[1]
     try:
         client_tile_upscaler = Client(tile_upscaler_url)
         print(f"Loaded custom model from {tile_upscaler_url}")
@@ -201,6 +183,24 @@ def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidanc
     return [image, result]
 
 
+def refine_image(apply_refiner, image, model ,prompt, negative_prompt, num_inference_steps, guidance_scale, seed, strength):
+    if (not apply_refiner):
+        return image
+    
+    client = InferenceClient()
+    refined_image = client.image_to_image(
+        handle_file(image), 
+        prompt=prompt, 
+        negative_prompt=negative_prompt, 
+        num_inference_steps=num_inference_steps, 
+        guidance_scale=guidance_scale, 
+        seed=seed, 
+        model=model,
+        strength=strength
+    )
+    return refined_image
+
+
 css="""
 """