Update app.py

app.py

@@ -11,7 +11,8 @@ def ocr(img,lang):
     # You can set the parameter `lang` as `ch`, `en`, `french`, `german`, `korean`, `japan`
     # to switch the language model in order.
     ocr = PaddleOCR(use_angle_cls=True, lang=lang) # need to run only once to download and load model into memory
-    img_path = 'PaddleOCR/doc/imgs_en/img_12.jpg'
+    print(img)
+    print(type(img))
     result = ocr.ocr(img, cls=True)
     for idx in range(len(result)):
         res = result[idx]
@@ -21,7 +22,7 @@ def ocr(img,lang):
     # draw result
     from PIL import Image
     result = result[0]
-    image = Image.open(img_path).convert('RGB')
+    image = img.convert('RGB')
     boxes = [line[0] for line in result]
     st.text(boxes)
     txts = [line[1][0] for line in result]
@@ -44,6 +45,6 @@ if submit:
         bytes_data = uploaded_file.getvalue()
         if bytes_data is not None:
             bytes_data = uploaded_file.read()
-            image = Image.open(bytes_data)
-            st.image(image, caption='Sunrise by the mountains')
+            image = Image.open(io.BytesIO(bytes_data))
+            st.image(image, caption='your img')
             ocr(image,'ch')