diff --git a/templates/generate.html b/templates/generate.html
index 2328ae7..de8f6b3 100644
--- a/templates/generate.html
+++ b/templates/generate.html
@@ -21,7 +21,7 @@
   <body class="bg-gray-50 text-gray-800">
     <div class="container mx-auto px-4">
       <nav class="flex justify-between items-center py-6">
-        <a href="#" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
+        <a href="/" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
         <div class="menu-toggle" onclick="toggleMenu()">
           <i class="fas fa-bars fa-2x"></i>
         </div>
diff --git a/templates/index.html b/templates/index.html
index b5d526d..5f7b384 100644
--- a/templates/index.html
+++ b/templates/index.html
@@ -26,7 +26,7 @@
 <body>
   <div class="container mx-auto px-4">
     <nav class="flex justify-between items-center py-6">
-      <a href="#" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
+      <a href="/" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
       <div class="menu-toggle" onclick="toggleMenu()">
         <i class="fas fa-bars fa-2x"></i>
       </div>
@@ -100,7 +100,7 @@ <h2 class="text-4xl font-bold text-gray-800 mb-4">Image Generator for Farm Servi
   <!-- Section 3: Image Generator -->
   <section class="flex flex-col md:flex-row justify-between items-center py-10">
     <div class="w-full md:w-1/2 mb-4 md:mb-0">
-        <img src="https://placehold.co/600x400" alt="Image Generator Visualization" class="mx-auto">
+        <img src="https://i.postimg.cc/vm8NbBH8/ed5289b6-b4fc-48fd-955d-c7014954b63f.webp" height="200" width="600" alt="Image Generator Visualization" class="mx-auto">
     </div>
     <div class="w-full md:w-1/2 px-8">
         <h2 class="text-4xl font-bold text-gray-800 mb-4">Comprehensive Business Model and Planning Tools</h2>
diff --git a/templates/predict.html b/templates/predict.html
index 1d93c38..7eb19db 100644
--- a/templates/predict.html
+++ b/templates/predict.html
@@ -18,7 +18,7 @@
 
 <!-- Navigation -->
 <nav class="flex justify-between items-center py-6">
-  <a href="#" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
+  <a href="/" class="text-2xl font-bold text-gray-800">Chalo Kisaan</a>
   <div class="menu-toggle" onclick="toggleMenu()">
     <i class="fas fa-bars fa-2x"></i>
   </div>
diff --git a/templates/visualize.html b/templates/visualize.html
index 173342c..988587d 100644
--- a/templates/visualize.html
+++ b/templates/visualize.html
@@ -2,37 +2,22 @@
 <html lang="en">
 <head>
     <meta charset="UTF-8">
-    <title>Image Generation</title>
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Work in Progress</title>
+    <link href="https://cdn.tailwindcss.com" rel="stylesheet">
+    <style>
+        /* Additional custom styles if needed */
+    </style>
 </head>
-<body>
-    <h2>Image Generation Form</h2>
-    <form id="generateForm" enctype="multipart/form-data">
-        <input type="file" id="image" name="image" required>
-        <button type="submit">Generate Image</button>
-    </form>
-    <h3>Generated Image:</h3>
-    <img id="outputImage" src="" alt="Generated Image" style="max-width: 500px;">
-
-    <script>
-        document.getElementById('generateForm').addEventListener('submit', function(e) {
-            e.preventDefault(); // Prevent the default form submission
-            
-            const formData = new FormData();
-            const imageInput = document.getElementById('image');
-            if (imageInput.files.length > 0) {
-                formData.append('image', imageInput.files[0]);
-            }
-
-            fetch('/generate-image', {
-                method: 'POST',
-                body: formData,
-            })
-            .then(response => response.json())
-            .then(data => {
-                document.getElementById('outputImage').src = data.generatedImageUrl;
-            })
-            .catch(error => console.error('Error:', error));
-        });
-    </script>
+<body class="bg-gray-100 flex items-center justify-center h-screen">
+    <div class="text-center">
+        <h1 class="text-4xl font-bold text-gray-800 mb-4">Work in Progress</h1>
+        <p class="text-lg text-gray-600 mb-5">We're working hard to finish the development of this page. Check back soon!</p>
+        <div>
+            <a href="/" class="inline-block bg-blue-500 text-white font-bold py-2 px-4 rounded-lg hover:bg-blue-600 transition duration-300">
+                Go Back to Home
+            </a>
+        </div>
+    </div>
 </body>
 </html>
