rbarman commited on
Commit
18229f3
·
1 Parent(s): 312de16

fix color format for output image

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -41,10 +41,10 @@ def predict(img: np.ndarray) -> str:
41
  #plt.imshow(result)
42
  print(f'result is: {type(result)}')
43
  print(result.shape)
44
- print(result)
45
 
46
  result_fp = 'temp_result.jpg'
47
- cv2.imwrite(result_fp, result)
48
  return result_fp
49
 
50
 
 
41
  #plt.imshow(result)
42
  print(f'result is: {type(result)}')
43
  print(result.shape)
44
+ #print(result)
45
 
46
  result_fp = 'temp_result.jpg'
47
+ cv2.imwrite(result_fp, cv2.cvtColor(result, cv2.cv2.COLOR_BGR2RGB))
48
  return result_fp
49
 
50