Update app.py

app.py

@@ -4,6 +4,7 @@ import time
 import torch
 import argparse
 import gradio as gr
+import io
 from PIL import Image
 from numpy import random
 from pathlib import Path
@@ -153,7 +154,7 @@ def detect_Custom(img,model,boundedImage):
     images = [keras_ocr.tools.read(img) for img in [boundedImage]]
     prediction_groups = pipeline.recognize(images)
     plt.figure(figsize = (10,20))
-    plt.imshow(images[0])
+    
     fig, ax = plt.subplots(figsize=(10, 20))
     keras_ocr.tools.drawAnnotations(image=images[0], predictions=prediction_groups[0], ax=ax)
     if save_txt or save_img:
@@ -161,7 +162,7 @@ def detect_Custom(img,model,boundedImage):
 
     print(f'Done. ({time.time() - t0:.3f}s)')
 
-    return Image.fromarray(im0[:,:,::-1]), asarray(images[0])
+    return Image.fromarray(im0[:,:,::-1]), keras_ocr.tools.drawAnnotations(image=images[0], predictions=prediction_groups[0], ax=ax))