Fix model path from models/model.pt to model.pt

app.py

@@ -26,7 +26,7 @@ def load_model():
         # Download model from HuggingFace
         model_path = hf_hub_download(
             repo_id="MONAI/example_spleen_segmentation",
-            filename="models/model.pt"
+            filename="model.pt"
         )
         
         # Initialize UNet architecture
@@ -105,7 +105,7 @@ def segment_spleen(input_file):
         overlay[seg_slice == 1] = [255, 0, 0]  # Red for spleen
         
         return overlay, output_path, "Segmentation completed successfully!"
-        
+    
     except Exception as e:
         return None, None, f"Error: {str(e)}"
 
@@ -115,13 +115,13 @@ with gr.Blocks(title="Spleen Segmentation") as demo:
     gr.Markdown(
         """Upload a CT scan in NIfTI format (.nii or .nii.gz) to segment the spleen using the 
         [MONAI/example_spleen_segmentation](https://huggingface.co/MONAI/example_spleen_segmentation) model.
-        
+
         **Model Info:**
         - Architecture: UNet
         - Input: 3D CT image (96×96×96)
         - Output: Binary segmentation (spleen vs background)
         - Mean Dice Score: 0.96
-        
+
         **Instructions:**
         1. Upload a NIfTI file (.nii or .nii.gz)
         2. Click Submit
@@ -155,7 +155,7 @@ with gr.Blocks(title="Spleen Segmentation") as demo:
         - nibabel
         - numpy
         - huggingface_hub
-        
+
         ### Citation
         If you use this model, please cite:
         ```