Fix WebUI not working inside of iframes

1 files changed, 2 insertions, 2 deletions

diff --git a/script.js b/script.js
index 8b3b67e3..d9424de9 100644
--- a/script.js
+++ b/script.js
@@ -1,5 +1,5 @@
 function gradioApp(){
-    return document.getElementsByTagName('gradio-app')[0].shadowRoot;
+    return document
 }
 
 function get_uiCurrentTab() {
@@ -82,4 +82,4 @@ function uiElementIsVisible(el) {
         }
     }
     return isVisible;
-}
\ No newline at end of file
+}