fix: use abort instead of execl

Dockerfile

@@ -79,4 +79,5 @@ COPY --chown=user . $HOME/app
 # -----------------------------------------------------------------------------
 USER user
 
-CMD ["python", "app.py"]
+# repeat cmd on abort
+CMD while python app.py; [ $? -eq 134 ]; do :; done

app.py

@@ -1,5 +1,5 @@
-from os import execl, getenv
-from sys import argv, exception, executable
+from os import abort, getenv
+from sys import exception
 from textwrap import dedent
 import traceback
 
@@ -11,13 +11,10 @@ from detikzify.webui import BANNER, build_ui, make_light
 def is_official_demo():
     return getenv("SPACE_AUTHOR_NAME") == "nllg"
 
-# Hack to temporarily work around memory leak, see:
-#   * https://huggingface.co/spaces/nllg/DeTikZify/discussions/2
-#   * https://github.com/gradio-app/gradio/issues/8503
-def reload_on_oom_hook(func):
+def add_abort_hook(func, *errors):
     def wrapper(*args, **kwargs):
-        if isinstance(exception(), (MemoryError, cuda.OutOfMemoryError)):
-            execl(executable, executable, *argv)
+        if isinstance(exception(), errors):
+            abort()
         return func(*args, **kwargs)
     return wrapper
 
@@ -37,7 +34,10 @@ else:
     use_big_models = cuda.is_available() and cuda.get_device_properties(0).total_memory > 15835660288
     model = f"detikzify-ds-{'7' if use_big_models else '1.3'}b"
     demo = build_ui(lock=is_official_demo(), model=model, light=True).queue()
-    traceback.print_exc = reload_on_oom_hook(traceback.print_exc)
+    # Hack to temporarily work around memory leak, see:
+    #   * https://huggingface.co/spaces/nllg/DeTikZify/discussions/2
+    #   * https://github.com/gradio-app/gradio/issues/8503
+    traceback.print_exc = add_abort_hook(traceback.print_exc, MemoryError, cuda.OutOfMemoryError)
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860)