Update app.py

app.py

@@ -5,6 +5,8 @@ from PIL import ImageDraw
 import gradio as gr
 import torch
 import easyocr
+import io
+import base64
 
 def draw_boxes(image, bounds, color='red', width=2):
     draw = ImageDraw.Draw(image)
@@ -19,14 +21,24 @@ def inference(img, lang):
     im = PIL.Image.open(img.name)
     draw_boxes(im, bounds)
     im.save('result.jpg')
+
+    #result_buffer = io.BytesIO()
+    
     return ['result.jpg', pd.DataFrame(bounds).iloc[: , 1:]]
 
 choices = ["en", "vi"] 
 
+#gr.Interface(
+#    inference,
+#    inputs=[gr.Image(label='Input'),
+#            gr.CheckboxGroup(choices, type="value", label='language')],
+#    outputs=[gr.Image(label='Output'), 
+#             gr.Dataframe(headers=['text', 'confidence'])]
+#    ).launch(debug=True)
+
 gr.Interface(
     inference,
     inputs=[gr.Image(label='Input'),
             gr.CheckboxGroup(choices, type="value", label='language')],
-    outputs=[gr.Image(label='Output'), 
-             gr.Dataframe(headers=['text', 'confidence'])]
+    outputs=gr.Image(label='Output')
     ).launch(debug=True)