Update app.py

app.py

@@ -310,6 +310,17 @@ css = """
     }
 """
 
+js = """
+var iv = setInterval(function() {
+    try {
+        if (BABYLON && BABYLON.Engine.LastCreatedScene.getNodes()[1].material.pointSize) {
+            alert(BABYLON.Engine.LastCreatedScene.getNodes()[1].material.pointSize);
+            clearInterval(iv);
+        }
+    } catch(e) {}
+}, 1000);
+"""
+
 title = "# Depth Anything Video Demo"
 description = """Depth Anything on full video files.  
 Please refer to our [paper](https://arxiv.org/abs/2401.10891), [project page](https://depth-anything.github.io), or [github](https://github.com/LiheYoung/Depth-Anything) for more details.  
@@ -333,7 +344,7 @@ transform = Compose([
 # def predict_depth(model, image):
 #     return model(image)
 
-with gr.Blocks(css=css) as demo:
+with gr.Blocks(css=css, js=js) as demo:
     gr.Markdown(title)
     gr.Markdown(description)
     gr.Markdown("### Video Depth Prediction demo")
@@ -513,17 +524,7 @@ with gr.Blocks(css=css) as demo:
         return output_video_path
 
     submit.click(on_submit, inputs=[input_video, model_type], outputs=[processed_video, processed_zip, output_frame, output_depth])
-    render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in], outputs=[result], js="""
-      var iv = setInterval(function() {
-        try {
-          if (BABYLON && BABYLON.Engine.LastCreatedScene.getNodes()[1].material.pointSize) {
-            alert(BABYLON.Engine.LastCreatedScene.getNodes()[1].material.pointSize);
-            clearInterval(iv);
-          }
-        } catch(e) {}
-        alert('...');
-      }, 1000);
-    """)
+    render.click(partial(get_mesh), inputs=[output_frame, output_depth, blur_in], outputs=[result])
 
     example_files = os.listdir('examples')
     example_files.sort()