Update templates/output.html

templates/output.html

@@ -159,12 +159,11 @@
 
     <script>
         function downloadImage() {
-            var link = document.createElement('a');
-            link.href = '/output_files/filename';
-            link.download = 'generated_image.jpg';
-            link.click();
-        }
-
+    var link = document.createElement('a');
+    link.href = '{{ url_for('download_output', filename=filename) }}';
+    link.download = 'generated_image.jpg';
+    link.click();
+}
         function toggleImageSize(image) {
             if (image.classList.contains('enlarged')) {
                 image.classList.remove('enlarged');