stalyn314 commited on
Commit
10b9fa1
·
verified ·
1 Parent(s): bfbcea8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def inference(image_url, image, min_score, model_name):
60
  r = requests.get(image_url)
61
  if r:
62
  im = np.frombuffer(r.content, dtype="uint8")
63
- im = cv2.imdecode(im, cv2.IMREAD_COLOR)
64
  else:
65
  # Model expect BGR!
66
  im = image[:,:,::-1]
 
60
  r = requests.get(image_url)
61
  if r:
62
  im = np.frombuffer(r.content, dtype="uint8")
63
+ im = cv2.imdecode(im, cv2.IMREAD_COLOR_BGR2RGB)
64
  else:
65
  # Model expect BGR!
66
  im = image[:,:,::-1]