feat: Update app.py to use new Tile-Upscaler client

app.py

@@ -13,17 +13,20 @@ job = None
 custom_model_url = "https://fffiloni-sd-xl-custom-model.hf.space"
 tile_upscaler_url = "https://gokaygokay-tileupscalerv2.hf.space"
 
-try:
-    client_custom_model = Client(custom_model_url)
-    print(f"Loaded custom model from {custom_model_url}")
-except ValueError as e:
-    print(f"Failed to load custom model: {e}")
+client_custom_model = None
+client_tile_upscaler = None
 
-try:
-    client_tile_upscaler = Client(tile_upscaler_url)
-    print(f"Loaded custom model from {tile_upscaler_url}")
-except ValueError as e:
-    print(f"Failed to load custom model: {e}")
+# try:
+#     client_custom_model = Client(custom_model_url)
+#     print(f"Loaded custom model from {custom_model_url}")
+# except ValueError as e:
+#     print(f"Failed to load custom model: {e}")
+
+# try:
+#     client_tile_upscaler = Client(tile_upscaler_url)
+#     print(f"Loaded custom model from {tile_upscaler_url}")
+# except ValueError as e:
+#     print(f"Failed to load custom model: {e}")
 
 
 def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width, height, seed, lora_weight, progress=gr.Progress(track_tqdm=True)):    
@@ -37,6 +40,29 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
         custom_model = selected_lora["repo"]
         trigger_word = selected_lora["trigger_word"]
 
+        global client_custom_model
+        global client_tile_upscaler
+
+        if client_custom_model is None:
+            try:
+                client_custom_model = Client(custom_model_url)
+                print(f"Loaded custom model from {custom_model_url}")
+            except ValueError as e:
+                print(f"Failed to load custom model: {e}")
+                client_custom_model = None
+                raise gr.Error("Failed to load client for " + custom_model_url)
+
+        if client_tile_upscaler is None:
+            try:
+                client_tile_upscaler = Client(tile_upscaler_url)
+                print(f"Loaded custom model from {tile_upscaler_url}")
+            except ValueError as e:
+                print(f"Failed to load custom model: {e}")
+                client_custom_model = None
+                raise gr.Error("Failed to load client for " + tile_upscaler_url)
+
+
+
         result = client_custom_model.submit(
             custom_model=custom_model,
             api_name="/load_model"