abdrabo01 commited on
Commit
abdc148
·
verified ·
1 Parent(s): b5754a0

Edit conf value

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -47,7 +47,7 @@ async def predict(request: ImageRequest):
47
  if image is None:
48
  return {"error": "Invalid image"}
49
 
50
- results = model.predict(image)
51
  result = results[0]
52
 
53
  json_result = {}
 
47
  if image is None:
48
  return {"error": "Invalid image"}
49
 
50
+ results = model.predict(image, conf=0.36)
51
  result = results[0]
52
 
53
  json_result = {}