Hugging Face's logo

Spaces:
svsaurav95
/
ToothSegmentation
Sleeping

App Files Community
svsaurav95 commited on
Commit
7b8b1ab
·
verified ·
1 Parent(s): 10e2895

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +1 -1
src/streamlit_app.py CHANGED
@@ -15,7 +15,7 @@ st.markdown("""
15
  </style>
16
  """, unsafe_allow_html=True)
17
 
18
- MODEL_PATH = mobilevit_teeth_segmentation.pth
19
 
20
  # === Model Definition ===
21
  class MobileViTSegmentation(nn.Module):
 
15
  </style>
16
  """, unsafe_allow_html=True)
17
 
18
+ MODEL_PATH = "mobilevit_teeth_segmentation.pth"
19
 
20
  # === Model Definition ===
21
  class MobileViTSegmentation(nn.Module):