feat: Handle exceptions when submitting requests to custom models and tile upscaler APIs

app.py

@@ -52,10 +52,14 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
                 raise gr.Error("Failed to load client for " + custom_model_url)
 
 
-        result = client_custom_model.submit(
-            custom_model=custom_model,
-            api_name="/load_model"
-        )
+        try:
+            result = client_custom_model.submit(
+                custom_model=custom_model,
+                api_name="/load_model"
+            )
+        except ValueError as e:
+            raise gr.Error(e)
+
         weight_name = result.result()[2]['value']
 
         if trigger_word and prompt.startswith(trigger_word):
@@ -67,19 +71,24 @@ def infer(selected_index, prompt, style_prompt, inf_steps, guidance_scale, width
         prompt_arr = [trigger_word, prompt, style_prompt]
         prompt = '. '.join([element.strip() for element in prompt_arr if element.strip() != ''])
         
-        job = client_custom_model.submit(
-            custom_model=custom_model,
-            weight_name=weight_name,
-            prompt=prompt,
-            inf_steps=inf_steps,
-            guidance_scale=guidance_scale,
-            width=width,
-            height=height,
-            seed=seed,
-            lora_weight=lora_weight,
-            api_name="/infer"
-        )
-        result = job.result()
+        try:
+            job = client_custom_model.submit(
+                custom_model=custom_model,
+                weight_name=weight_name,
+                prompt=prompt,
+                inf_steps=inf_steps,
+                guidance_scale=guidance_scale,
+                width=width,
+                height=height,
+                seed=seed,
+                lora_weight=lora_weight,
+                api_name="/infer"
+            )
+            result = job.result()
+        except ValueError as e:
+            raise gr.Error(e)
+
+
         new_result = result + (prompt, )
 
         return new_result
@@ -117,17 +126,21 @@ def upscale_image(image, resolution, num_inference_steps, strength, hdr, guidanc
             client_custom_model = None
             raise gr.Error("Failed to load client for " + tile_upscaler_url)
 
-    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"
-    )
+    try:
+        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"
+        )
+    except ValueError as e:
+            raise gr.Error(e)
+
     return result