Fix: Set SegResNet init_filters=32 to match checkpoint

app.py

@@ -63,7 +63,7 @@ def load_model():
     model = SegResNet(
         blocks_down=[1, 2, 2, 4],
         blocks_up=[1, 1, 1],
-        init_filters=16,
+        init_filters=32,
         in_channels=1,
         out_channels=105,
         dropout_prob=0.2,
@@ -294,10 +294,12 @@ with gr.Blocks(
             
             # Example files
             gr.Markdown("### 📁 Example Files")
+            
+            # Dynamically list all .nii.gz files in examples folder
+            example_files = [[os.path.join("examples", f)] for f in os.listdir("examples") if f.endswith(".nii.gz")]
+            
             example_gallery = gr.Examples(
-                examples=[
-                    ["examples/sample_ct_real.nii.gz"],
-                ],
+                examples=example_files,
                 inputs=[file_input],
                 label="Click to load example"
             )