feat: Refactor upscale function to use new Tile-Upscaler client

app.py

@@ -84,22 +84,22 @@ def update_selection(evt: gr.SelectData):
     )
 
 
-def upscale(image, resolution, inf_steps, strength, hdr_effect, guidance_scale):
-    try:
-        # other_client = Client("gokaygokay/Tile-Upscaler")
-        # result = other_client.predict(
-        #     param_0=handle_file(image),
-        #     param_1=resolution,
-        #     param_2=inf_steps,
-        #     param_3=strength,
-        #     param_4=hdr_effect,
-        #     param_5=guidance_scale,
-        #     api_name="/wrapper"
-        # )
-        return [image, image]
-        return result
-    except Exception as e:
-        gr.Warning("Error: " + str(e))
+def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidance_scale, controlnet_strength, scheduler_name):
+  result = client_tile_upscaler.predict(
+      param_0=handle_file(image),
+      param_1=resolution,
+      param_2=num_inference_steps,
+      param_3=strength,
+      param_4=hdr,
+      param_5=guidance_scale,
+      param_6=controlnet_strength,
+      param_7=scheduler_name,
+      api_name="/wrapper"
+  )
+  return result
+
+
+
 
 css="""
 """
@@ -220,7 +220,7 @@ with gr.Blocks(css=css) as demo:
     )
 
     def upscale_with_fixed_values(image):
-        return upscale(image, 768, 25, 0.4, 0.3, 7.5)
+        return upscale_image(image, 768, 25, 0.4, 0.3, 7.5)
 
     scale_btn.click(
         fn=upscale_with_fixed_values,