Update app.py

app.py

@@ -47,8 +47,8 @@ def visualize_slice(image, mask, slice_index):
     fig, axes = plt.subplots(1, 2, figsize=(12, 6))
 
     # Extract the 2D slices
-    image_slice = image[:, :, slice_index]
-    mask_slice = mask[:, :, slice_index]
+    image_slice = image[slice_index, :, :]
+    mask_slice = mask[slice_index, :, :]
 
     # Plot image slice
     axes[0].imshow(image_slice, cmap="gray")
@@ -76,14 +76,15 @@ iface = gr.Interface(
     fn=interface,
     inputs=[
         gr.File(label="Upload 3D Image (.tif or .czi)"),
-        gr.Slider(minimum=0, maximum=100, step=1, label="Slice Index", live=True)  # Set live=True for dynamic updates
+        gr.Slider(minimum=0, maximum=100, step=1, label="Slice Index")  # No live=True
     ],
     outputs=[
         gr.Plot(label="2D Slice Visualization"),
         gr.File(label="Download Binary Mask (.tif)")
     ],
     title="3D Image Processing with Binary Mask Output",
-    description="Upload a 3D image in .tif or .czi format. The model will process the image and output a 3D binary mask. Use the slider to navigate through the 2D slices."
+    description="Upload a 3D image in .tif or .czi format. The model will process the image and output a 3D binary mask. Use the slider to navigate through the 2D slices.",
+    live=True  # Enable real-time updates in Interface
 )
 
 if __name__ == "__main__":