phuochungus commited on
Commit
a5c51fb
1 Parent(s): 1fd5f60

Update threshold values in image and video routers

Browse files
Files changed (2) hide show
  1. app/routers/image.py +1 -1
  2. app/routers/video.py +5 -3
app/routers/image.py CHANGED
@@ -11,7 +11,7 @@ router = APIRouter(prefix="/image", tags=["Image"])
11
  @router.post("")
12
  async def handleImageRequest(
13
  file: bytes = File(...),
14
- threshold: float = 0.3,
15
  ):
16
  try:
17
  img = imfrombytes(file, cv2.IMREAD_COLOR)
 
11
  @router.post("")
12
  async def handleImageRequest(
13
  file: bytes = File(...),
14
+ threshold: float = 0.5,
15
  ):
16
  try:
17
  img = imfrombytes(file, cv2.IMREAD_COLOR)
app/routers/video.py CHANGED
@@ -32,8 +32,8 @@ router = APIRouter(prefix="/video", tags=["Video"])
32
  async def handleVideoRequest(
33
  file: UploadFile,
34
  background_tasks: BackgroundTasks,
35
- threshold: float = 0.3,
36
- user=Depends(get_current_user)
37
  ):
38
  if re.search("^video\/", file.content_type) is None:
39
  raise HTTPException(
@@ -46,7 +46,9 @@ async def handleVideoRequest(
46
  _, artifact_ref = db.collection("artifacts").add(
47
  {"name": id + ".mp4", "status": "pending"}
48
  )
49
- db.collection("user").document(user["sub"]).update({"artifacts": ArrayUnion(['artifact/' + artifact_ref.id])})
 
 
50
  os.mkdir(id)
51
  async with aiofiles.open(os.path.join(id, "input.mp4"), "wb") as out_file:
52
  while content := await file.read(102400):
 
32
  async def handleVideoRequest(
33
  file: UploadFile,
34
  background_tasks: BackgroundTasks,
35
+ threshold: float = 0.5,
36
+ user=Depends(get_current_user),
37
  ):
38
  if re.search("^video\/", file.content_type) is None:
39
  raise HTTPException(
 
46
  _, artifact_ref = db.collection("artifacts").add(
47
  {"name": id + ".mp4", "status": "pending"}
48
  )
49
+ db.collection("user").document(user["sub"]).update(
50
+ {"artifacts": ArrayUnion(["artifact/" + artifact_ref.id])}
51
+ )
52
  os.mkdir(id)
53
  async with aiofiles.open(os.path.join(id, "input.mp4"), "wb") as out_file:
54
  while content := await file.read(102400):