hantech commited on
Commit
b2a4c2e
β€’
1 Parent(s): 197d458

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -48,7 +48,9 @@ def inference(filepath, lang):
48
  # crop the region of interest (ROI)
49
 
50
  cropped_image = img[min_x:(max_x - min_x), min_y:(max_y - min_y)] # crop the image
51
-
 
 
52
  out = recognitor.predict(cropped_image)
53
  print(out)
54
  new_bounds.append((bbox,text, out, prob))
 
48
  # crop the region of interest (ROI)
49
 
50
  cropped_image = img[min_x:(max_x - min_x), min_y:(max_y - min_y)] # crop the image
51
+ cropped_image = Image.fromarray(cropped_image)
52
+
53
+ cropped_image = cropped_image.convert("RGB")
54
  out = recognitor.predict(cropped_image)
55
  print(out)
56
  new_bounds.append((bbox,text, out, prob))