Update src/streamlit_app.py

src/streamlit_app.py

@@ -43,7 +43,7 @@ class MobileViTSegmentation(nn.Module):
 @st.cache_resource
 def load_model():
     model = MobileViTSegmentation()
-    state_dict = torch.load("mobilevit_teeth_segmentation.pth", map_location="cpu")
+    state_dict = torch.load("ToothSegmentation/mobilevit_teeth_segmentation.pth", map_location="cpu")
     model.load_state_dict(state_dict)
     model.eval()
     return model