Spaces:

Noursine
/

Teeth_segmentation_annotation_instance_seg

Runtime error

Noursine commited on Jun 13, 2025

Commit

f8de414

verified ·

1 Parent(s): d6779b2

Update src/streamlit_app.py

Files changed (1) hide show

src/streamlit_app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from ultralytics import YOLO
 @st.cache_resource(show_spinner=False)
 def load_model():
-    return YOLO('/src/best.pt')
 def predict(image, model):
     img_np = np.array(image)

 @st.cache_resource(show_spinner=False)
 def load_model():
+    return YOLO('/app/src/best.pt')
 def predict(image, model):
     img_np = np.array(image)