show trace as gradio app. lmao smh.

app.py

@@ -23,8 +23,12 @@ _model = InferenceUNetPseudo3D(
         hf_auth_token = os.environ.get('HUGGING_FACE_HUB_TOKEN', None)
 )
 
-if _model.failed == True:
-    exit()
+if _model.failed != False:
+    trace = f'```{_model.failed}```'
+    with gr.Blocks(title = 'Make-A-Video Stable Diffusion JAX', analytics_enabled = False) as demo:
+        exception = gr.Markdown(trace)
+
+    demo.launch()
 
 # gradio is illiterate. type hints make it go poopoo in pantsu.
 def generate(

makeavid_sd/inference.py

@@ -84,8 +84,8 @@ class InferenceUNetPseudo3D:
             del unet_params
         except Exception as e:
             print(e)
+            self.failed = ''.join(traceback.format_exception(None, e, e.__traceback__))
             traceback.print_exc()
-            self.failed = True
             return
         self.failed = False
         vae, vae_params = FlaxAutoencoderKL.from_pretrained(