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,7 +76,7 @@ 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")  # Adjust max value based on image size
+        gr.Slider(minimum=0, maximum=100, step=1, label="Slice Index", live=True)  # Set live=True for dynamic updates
     ],
     outputs=[
         gr.Plot(label="2D Slice Visualization"),