Update static/index.html

static/index.html

@@ -239,7 +239,8 @@ h1 {
  
   
     <a id="download-link" href="#" download="colorized_image.jpg" style="display:none;">Download Colorized Image</a>
-    
+    </div>
+    </div>
     <script>
     	 function previewImage(event, imageId) {
             const fileInput = event.target;
@@ -254,6 +255,18 @@ h1 {
             reader.readAsDataURL(file);
         }
         
+        
+         function displayColorizedImage(blob) {
+            const url = URL.createObjectURL(blob);
+          const resultContainer = document.getElementById('resultContainer');
+            resultContainer.innerHTML = `<img src="${url}" alt="Colorized Image" width="400">`;
+            const downloadLink = document.getElementById('download-link');
+            downloadLink.href = url;
+            downloadLink.style.display = 'block';
+        }
+        
+        
+        
         async function uploadFile() {
             const sourceFileInput = document.getElementById('sourceFile');
            const resultContainer = document.getElementById('resultContainer');
@@ -289,8 +302,7 @@ h1 {
             // If successful response, display the result image
               const blob = await response.blob();
             
-            displayColorizedImage(blob)
-            resultContainer.appendChild(blob);
+            displayColorizedImage(blob);
 
 
         } else {