Update app.py

app.py

@@ -9,14 +9,21 @@ import easyocr
 @spaces.GPU
 def process(img_path):
     reader = easyocr.Reader(['en'])
-    bounds = reader.readtext(img_path)
-
-    return [img_path, pd.DataFrame(bounds).iloc[0,1]]
+    results = reader.readtext(img_path)
+    ret = ""
+    pim = PIL.Image.open(img_path)
+    draw = ImageDraw.Draw(pim)
+    for (bbox, text, prob) in result:
+        p0, p1, p2, p3 = bbox
+        draw.line([*p0, *p1, *p2, *p3, *p0], fill = 'yellow', width = 2)
+        ret = text
+        break
+    return [img_path, ret]
 
 title = '车架号识别展示'
 description = '拖拽图片到识别区开始分析'
 ip = gr.Image(sources = ['upload'], label = '上传图片', width = '100%', height = '40rem', type = 'filepath')
-op = [gr.Image(sources = None, label = '处理结果', width = '100%', height = '40rem', type = 'filepath'), gr.Textbox()]
+op = [gr.Image(sources = None, label = '处理结果', width = '100%', height = '40rem', type = 'filepath'), gr.Textbox(show_label = False)]
 css = """
 table.svelte-y11bhb.svelte-y11bhb { max-width: 100%; overflow-x: hidden; }
 thead.svelte-y11bhb.svelte-y11bhb { visibility: hidden; }