rbarman commited on
Commit
ea854e0
·
1 Parent(s): fe355da

use input image!

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def predict(img: np.ndarray) -> str:
20
  # input: numpy array of image in RGB (see defaults for https://www.gradio.app/docs/#image)
21
 
22
  # Text detection models expect an image in BGR format.
23
- image = cv2.cvtColor(im_cv, cv2.COLOR_RGB2BGR)
24
  # N,C,H,W = batch size, number of channels, height, width.
25
  N, C, H, W = input_layer_ir.shape
26
  # Resize the image to meet network expected input sizes.
 
20
  # input: numpy array of image in RGB (see defaults for https://www.gradio.app/docs/#image)
21
 
22
  # Text detection models expect an image in BGR format.
23
+ image = cv2.cvtColor(img, cv2.COLOR_RGB2BGR)
24
  # N,C,H,W = batch size, number of channels, height, width.
25
  N, C, H, W = input_layer_ir.shape
26
  # Resize the image to meet network expected input sizes.