Update src/streamlit_app.py

src/streamlit_app.py

@@ -6,7 +6,7 @@ from ultralytics import YOLO
 
 @st.cache_resource(show_spinner=False)
 def load_model():
-    return YOLO('best.pt')  # Make sure the model file is in the same folder or provide full path
+    return YOLO('best.pt') 
 
 def predict(image, model):
     img_np = np.array(image)
@@ -83,4 +83,7 @@ def main():
                 st.write(f"Tooth {i+1}: {label}")
 
         else:
-            st.info("No teeth detected in this image.")
+            st.info("No teeth detected in this image.")
+
+if __name__ == "__main__":
+    main()