Update app.py

app.py

@@ -48,6 +48,9 @@ def process_image(input_image):
     output_image = prediction[0] * 0.5 + 0.5  # Denormalize to [0, 1]
     output_image = tf.clip_by_value(output_image, 0, 1)
 
+    # Resize the output image to the desired resolution
+    output_image = tf.image.resize(output_image, output_resolution, method='bilinear')
+
     # Convert to numpy array and then to PIL Image
     output_array = (output_image.numpy() * 255).astype(np.uint8)
     output_pil = Image.fromarray(output_array)