phuochungus commited on
Commit
ae4bf0f
1 Parent(s): 2cf679a

Refactor image inference function name

Browse files
Files changed (1) hide show
  1. app/routers/video.py +2 -2
app/routers/video.py CHANGED
@@ -21,7 +21,7 @@ from firebase_admin import messaging
21
  from app import db
22
  from app import supabase
23
  from app.dependencies import get_current_user
24
- from app.routers.image import inferenceImage
25
  from google.cloud.firestore_v1.base_query import FieldFilter
26
  from google.cloud.firestore import ArrayUnion
27
  from app import logger
@@ -99,7 +99,7 @@ def inferenceFrame(inputDir, threshold: float = 0.3):
99
  if res == False:
100
  break
101
 
102
- resFram = inferenceImage(frame, threshold, False)
103
  result.write(resFram)
104
  cap.release()
105
  result.release()
 
21
  from app import db
22
  from app import supabase
23
  from app.dependencies import get_current_user
24
+ from app.routers.image import inference_image
25
  from google.cloud.firestore_v1.base_query import FieldFilter
26
  from google.cloud.firestore import ArrayUnion
27
  from app import logger
 
99
  if res == False:
100
  break
101
 
102
+ resFram = inference_image(frame, threshold)
103
  result.write(resFram)
104
  cap.release()
105
  result.release()