Update app.py

app.py

@@ -30,9 +30,12 @@ loaded_autoencoder = tf.keras.models.load_model(
 )
 
 def process_image(input_img):
-    # Convert PIL Image to grayscale and resize
+    # Store original input dimensions
+    original_width, original_height = input_img.size
+
+    # Convert PIL Image to grayscale and resize to model input size
     img = input_img.convert("L")  # Convert to grayscale (single channel)
-    img = img.resize((WIDTH, HEIGHT))  # Resize to 512x512
+    img = img.resize((WIDTH, HEIGHT))  # Resize to 512x512 for model
     img_array = tf.keras.preprocessing.image.img_to_array(img) / 255.0  # Normalize to [0, 1]
     img_array = img_array[None, ..., 0:1]  # Add batch dimension, shape: (1, 512, 512, 1)
 
@@ -52,15 +55,18 @@ def process_image(input_img):
     rgb_array = np.clip(rgb_array, 0, 1) * 255.0  # Scale to [0, 255]
     rgb_image = Image.fromarray(rgb_array.astype(np.uint8), mode="RGB")  # Create RGB PIL image
 
+    # Resize output image to match input image resolution
+    rgb_image = rgb_image.resize((original_width, original_height), Image.Resampling.LANCZOS)
+
     return rgb_image
 
 custom_css = """
-body {background: linear-gradient(135deg, #232526 0%, #414345 100%) !important;}
+body {background: linear-gradient(135deg, #f0f4f8 0%, #d9e2ec 100%) !important;}
 .gradio-container {background: transparent !important;}
-h1, .gr-title {color: #00e6d3 !important; font-family: 'Segoe UI', sans-serif;}
-.gr-description {color: #e0e0e0 !important; font-size: 1.1em;}
-.gr-input, .gr-output {border-radius: 18px !important; box-shadow: 0 4px 24px rgba(0,0,0,0.18);}
-.gr-button {background: linear-gradient(90deg, #00e6d3 0%, #0072ff 100%) !important; color: #fff !important; border: none !important; border-radius: 12px !important;}
+h1, .gr-title {color: #007bff !important; font-family: 'Segoe UI', sans-serif;}
+.gr-description {color: #333333 !important; font-size: 1.1em;}
+.gr-input, .gr-output {border-radius: 18px !important; box-shadow: 0 4px 24px rgba(0,0,0,0.1);}
+.gr-button {background: linear-gradient(90deg, #007bff 0%, #00c4cc 100%) !important; color: #fff !important; border: none !important; border-radius: 12px !important;}
 """
 
 demo = gr.Interface(
@@ -78,8 +84,8 @@ demo = gr.Interface(
     css=custom_css,
     allow_flagging="never",
     examples=[
-        ["examples/example_1.jpg"],
-        ["examples/example_2.jpg"]
+        ["examples/example_input_1.jpg"],
+        ["examples/example_input_2.jpg"]
     ]
 )