1 files changed, 17 insertions, 0 deletions
diff --git a/script.js b/script.js
index ff301e49..86ec3da6 100644
--- a/script.js
+++ b/script.js
@@ -79,6 +79,23 @@ function addTitles(root){
 	    global_progressbar = progressbar
 
         var mutationObserver = new MutationObserver(function(m){
+            txt2img_preview = gradioApp().getElementById('txt2img_preview')
+            txt2img_gallery = gradioApp().getElementById('txt2img_gallery')
+
+            img2img_preview = gradioApp().getElementById('img2img_preview')
+            img2img_gallery = gradioApp().getElementById('img2img_gallery')
+
+            if(txt2img_preview != null && txt2img_gallery != null){
+                txt2img_preview.style.width = txt2img_gallery.clientWidth + "px"
+                txt2img_preview.style.height = txt2img_gallery.clientHeight + "px"
+            }
+
+            if(img2img_preview != null && img2img_gallery != null){
+                img2img_preview.style.width = img2img_gallery.clientWidth + "px"
+                img2img_preview.style.height = img2img_gallery.clientHeight + "px"
+            }
+
+
             window.setTimeout(requestProgress, 500)
         });
         mutationObserver.observe( progressbar, { childList:true, subtree:true })