add CUDA error trigger

tools/webui/inference.py

@@ -42,15 +42,21 @@ def inference_wrapper(
         seed=int(seed) if seed else None,
         use_memory_cache=use_memory_cache,
     )
-
-    for result in engine.inference(req):
-        match result.code:
-            case "final":
-                return result.audio, None
-            case "error":
-                return None, build_html_error_message(i18n(result.error))
-            case _:
-                pass
+    try:
+        for result in engine.inference(req):
+            match result.code:
+                case "final":
+                    return result.audio, None
+                case "error":
+                    return None, build_html_error_message(i18n(result.error))
+                case _:
+                    pass
+    except Exception as e:
+        er = "CUDA error: device-side assert triggered"
+        if er in str(e):
+            app.close()
+        else:
+            raise Exception(e)
 
     return None, i18n("No audio generated")