Intelligent-Text-to-Image

Runtime error

testmail-gmail commited on Mar 21, 2023

Commit

81c81bb

1 Parent(s): b779888

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -153,7 +153,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))
     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:

     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: