Update app.py

app.py

@@ -8,13 +8,13 @@ from PIL import Image
 # Set Matplotlib to use the 'Agg' backend
 plt.switch_backend('Agg')
 
-def load_nrrd(file_path):
-    data, _ = nrrd.read(file_path.name)
+def load_nrrd(file_obj):
+    data, _ = nrrd.read(file_obj)
     num_slices = data.shape[2]
     return data, num_slices
 
-def visualize_slice(file_path, slice_index):
-    data, num_slices = load_nrrd(file_path)
+def visualize_slice(file_obj, slice_index):
+    data, num_slices = load_nrrd(file_obj)
     
     # Ensure the slice index is within the range of available slices
     slice_index = min(max(0, slice_index), num_slices - 1)
@@ -35,8 +35,8 @@ def visualize_slice(file_path, slice_index):
     
     return pil_img
 
-def update_slider(file_path):
-    _, num_slices = load_nrrd(file_path)
+def update_slider(file_obj):
+    _, num_slices = load_nrrd(file_obj)
     return gr.update(maximum=num_slices-1, value=0)
 
 with gr.Blocks() as app:
@@ -51,4 +51,4 @@ with gr.Blocks() as app:
     file_input.change(fn=visualize_slice, inputs=[file_input, slider], outputs=image_output)
     slider.change(fn=visualize_slice, inputs=[file_input, slider], outputs=image_output)
 
-app.launch()
+app.launch()