summaryrefslogtreecommitdiff
path: root/client/src/pages/Enrollment/Enrollment.jsx
diff options
context:
space:
mode:
Diffstat (limited to 'client/src/pages/Enrollment/Enrollment.jsx')
-rw-r--r--client/src/pages/Enrollment/Enrollment.jsx6
1 files changed, 5 insertions, 1 deletions
diff --git a/client/src/pages/Enrollment/Enrollment.jsx b/client/src/pages/Enrollment/Enrollment.jsx
index 35a599f..4d8fad2 100644
--- a/client/src/pages/Enrollment/Enrollment.jsx
+++ b/client/src/pages/Enrollment/Enrollment.jsx
@@ -28,6 +28,7 @@ import BackButton from '../../components/BackButton/BackButton'
import { initialUserData } from '../../constants/userData'
import * as cocoSsd from '@tensorflow-models/coco-ssd'
import * as tf from '@tensorflow/tfjs'
+import Spinner from '../../components/Spinner/Spinner'
const Enrollment = () => {
const { t } = useTranslation()
@@ -73,6 +74,9 @@ const Enrollment = () => {
})
setPage(page + 1)
},
+ onLoading: () => {
+ return <Spinner heading='ENROLLMENT' />
+ },
onError: (error) => {
console.log(error)
if (error?.response?.data?.message === 'User already exists.') {
@@ -152,7 +156,7 @@ const Enrollment = () => {
toast.error(t('PLEASE_CAPTURE_PHOTOGRAPH'))
} else if (
predictions[0].class === 'person' &&
- predictions[0].score > 0.8
+ predictions[0].score > 0.7
) {
setPage(page + 1)
} else {