Update app.py

app.py

@@ -6,30 +6,26 @@ from PIL import ImageDraw
 import gradio as gr
 import easyocr
 
-def draw_boxes(image, bounds, color='yellow', width = 2):
-    draw = ImageDraw.Draw(image)
-    for bound in bounds:
-        p0, p1, p2, p3 = bound[0]
-        draw.line([*p0, *p1, *p2, *p3, *p0], fill = color, width = width)
-    return image
-
 @spaces.GPU
-def inference(img):
+def process(img):
     reader = easyocr.Reader(['en'])
     bounds = reader.readtext(img.name)
     im = PIL.Image.open(img.name)
-    draw_boxes(im, bounds)
+    draw = ImageDraw.Draw(image)
+    for bound in bounds:
+        p0, p1, p2, p3 = bound[0]
+        draw.line([*p0, *p1, *p2, *p3, *p0], fill = 'yellow', width = 2)
     im.save('result.jpg')
     return ['result.jpg', pd.DataFrame(bounds).iloc[: , 1:]]
 
 title = '车架号识别展示'
 description = '拖拽图片到识别区开始分析'
-ip = [gr.Image(sources = ['upload'], label = '上传图片', width = "100%", height = "40rem")]
+ip = gr.Image(sources = ['upload'], label = '上传图片', width = "100%", height = "40rem")
 op = [gr.Image(sources = [], label = '处理结果', width = "100%", height = "40rem"), gr.Dataframe(col_count = (1, 'fixed'), show_label = False)]
 css = """
 table.svelte-y11bhb.svelte-y11bhb { max-width: 100%; overflow-x: hidden; }
 thead.svelte-y11bhb.svelte-y11bhb { visibility: hidden; }
 #component-2 div.progress-text.svelte-ls20lj.meta-text { visibility: hidden; } 
 """
-gif = gr.Interface(fn = inference, inputs = ip, outputs = op, title = title, description = description, batch = False, css = css, flagging_mode='never')
+gif = gr.Interface(fn = process, inputs = ip, outputs = op, title = title, description = description, batch = False, css = css, flagging_mode='never')
 gif.launch()