Update src/streamlit_app.py

src/streamlit_app.py

@@ -36,9 +36,17 @@ class MobileViTSegmentation(nn.Module):
 # ========== Load Model ==========
 @st.cache_resource
 def load_model():
-    checkpoint_path = hf_hub_download(repo_id="svsaurav95/ToothSegmentation", filename="mobilevit_teeth_segmentation.pth")
+    from model import MobileViTSegmentation  # or define it here if needed
+    cache_dir = "/tmp/huggingface"  # Writable path in HF Spaces
+
+    checkpoint_path = hf_hub_download(
+        repo_id="svsaurav95/ToothSegmentation",
+        filename="mobilevit_teeth_segmentation.pth",
+        cache_dir=cache_dir  # <- this fixes the permission issue
+    )
+
     model = MobileViTSegmentation(pretrained=False)
-    model.load_state_dict(torch.load(checkpoint_path, map_location='cpu'))
+    model.load_state_dict(torch.load(checkpoint_path, map_location="cpu"))
     model.eval()
     return model