Update app.py

app.py

@@ -116,6 +116,7 @@ def add_gallery(image, model_str, gallery):
 js="""
 <script>
 
+// Function to download an image
 downloadImage = (url, filename) => {
     const a = document.createElement('a');
     a.href = url;
@@ -149,12 +150,16 @@ const monitorImageChange = (imageElement) => {
     console.log('Now monitoring image changes for:', imageElement);
 };
 
-const img = document.querySelector('img'); // Select the target image element
-if (img) {
-    monitorImageChange(img);
-} else {
-    console.error('No image found to monitor.');
-}
+// Start monitoring after page load
+window.addEventListener('load', () => {
+    const img = document.querySelector('img'); // Select the target image element
+    if (img) {
+        monitorImageChange(img);
+    } else {
+        console.error('No image found to monitor.');
+    }
+});
+
     
 </script>
 """