js optimizations

javascript/imageMaskFix.js

@@ -1,4 +1,3 @@
-/* global gradioApp, onUiUpdate */
 /**
  * temporary fix for https://github.com/AUTOMATIC1111/stable-diffusion-webui/issues/668
  * @see https://github.com/gradio-app/gradio/issues/1721
@@ -41,5 +40,5 @@ function imageMaskResize() {
   });
 }
 
-onUiUpdate(imageMaskResize);
+onAfterUiUpdate(imageMaskResize);
 window.addEventListener('resize', imageMaskResize);