Update app.py

app.py

@@ -551,11 +551,11 @@ def draw_mask(l, t, v, d, evt: gr.EventData):
 
 
 load_model="""
-async(f, c, o, b, p, d, n)=>{
+async(c, o, b, p, d, n)=>{
   var intv = setInterval(function(){
-  try {
     if (document.getElementById("iframe3D")===null || typeof document.getElementById("iframe3D")==="undefined") {
-      if (typeof BABYLON !== "undefined" && BABYLON.Engine && BABYLON.Engine.LastCreatedScene) {
+      try {
+      if (BABYLON && BABYLON.Engine && BABYLON.Engine.LastCreatedScene) {
         BABYLON.Engine.LastCreatedScene.onAfterRenderObservable.add(function() { //onDataLoadedObservable
 
           var then = new Date().getTime();
@@ -619,11 +619,12 @@ async(f, c, o, b, p, d, n)=>{
         
         clearInterval(intv);
       }
+      } catch(e) {alert(e);}
     } else if (BABYLON) {
       BABYLON = null;
       document.getElementById("model3D").getElementsByTagName("canvas")[0].remove();
       
-      document.getElementById("iframe3D").src = "index.htm?file=" + f.path;
+      document.getElementById("iframe3D").src = "index.htm";
       document.getElementById("iframe3D").onload = function() {
         document.getElementById("iframe3D").contentDocument.getElementById("coords").value = c;
         document.getElementById("iframe3D").contentDocument.getElementById("order").value = o;
@@ -635,7 +636,6 @@ async(f, c, o, b, p, d, n)=>{
       
       clearInterval(intv);
     }
-  } catch(e) {alert(e);}
   }, 40);
 }
 """
@@ -1156,8 +1156,8 @@ with gr.Blocks(css=css, js=js) as demo:
         return output_video_path + (json.dumps(locations),)
 
     submit.click(on_submit, inputs=[input_video, model_type, coords], outputs=[processed_video, processed_zip, output_frame, output_mask, coords])
+    render.click(None, inputs=[coords, mesh_order, bgcolor, output_frame, output_mask, selected], outputs=None, js=load_model)
     render.click(partial(get_mesh), inputs=[output_frame, output_mask, blur_in, load_all], outputs=[result, result_file, mesh_order])
-    result_file.change(None, inputs=[result_file, coords, mesh_order, bgcolor, output_frame, output_mask, selected], outputs=None, js=load_model)
 
     example_files = os.listdir('examples')
     example_files.sort()