Make the generated image count only count new images in the currently active tab

3 years ago · 303df25cc2
parent 685f9631b5
commit 303df25cc2
1 changed files with 1 additions and 1 deletions
--- a/javascript/notification.js
+++ b/javascript/notification.js
@ -15,7 +15,7 @@ onUiUpdate(function(){
        }
    }

-    const galleryPreviews = gradioApp().querySelectorAll('img.h-full.w-full.overflow-hidden');
+    const galleryPreviews = gradioApp().querySelectorAll('div[id^="tab_"][style*="display: block"] img.h-full.w-full.overflow-hidden');

    if (galleryPreviews == null) return;