Update AB/index.html

AB/index.html

@@ -256,17 +256,18 @@ h1 {
         }
         
         
-         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';
+         // Function to display the processed image
+        function displayResultImage(url) {
+            const resultContainer = document.getElementById('resultContainer');
+            // Create an image element
+            const img = document.createElement('img');
+            // Set the source attribute to the processed image URL
+            img.src = url;
+            // Append the image element to the result container
+            resultContainer.appendChild(img);
         }
         
         
-        
         async function uploadFile() {
             const sourceFileInput = document.getElementById('sourceFile');
            const resultContainer = document.getElementById('resultContainer');
@@ -302,7 +303,7 @@ h1 {
             // If successful response, display the result image
               const blob = await response.blob();
             
-            displayColorizedImage(blob);
+            displayResultImage(url);
 
 
         } else {
@@ -331,3 +332,4 @@ h1 {
     </script>
 </body>
 </html>
+