Update app.py

app.py

@@ -22,7 +22,9 @@ def ocr(img,lang):
     # draw result
     from PIL import Image
     result = result[0]
-    image = img.convert('RGB')
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    im_pil = Image.fromarray(img)
+    image = im_pil
     boxes = [line[0] for line in result]
     st.text(boxes)
     txts = [line[1][0] for line in result]
@@ -31,7 +33,7 @@ def ocr(img,lang):
     st.text(scores)
     im_show = draw_ocr(image, boxes, txts, scores, font_path='/simfang.ttf')
     im_show = Image.fromarray(im_show)
-    st.image(im_show, caption='Sunrise by the mountains')
+    st.image(im_show, caption='ocr output')
     #im_show.save('result.jpg')
 
 _, col0, _ = st.columns([1, 3, 1])