Update app.py

app.py

@@ -11,7 +11,10 @@ import math
 from numpy import asarray
 import numpy as np
 
-torch.hub.download_url_to_file('https://i.pinimg.com/originals/45/d0/30/45d03054e15f4be731781eecba7458a4.jpg', 'korean.jpg')
+torch.hub.download_url_to_file('https://i.pinimg.com/736x/93/d3/54/93d354497ea26d5dc181055b9356cf79.jpg', 'korean.jpg')
+torch.hub.download_url_to_file('https://64.media.tumblr.com/1a5796817934a179664508693cae82d3/67c2ea7948c385cc-1b/s1280x1920/a70d4fcc0a5528de415b024db4ee6036da9c4c35.jpg', 'chinese.jpg')
+torch.hub.download_url_to_file('https://digistatement.com/wp-content/uploads/2020/06/12a7c208-33bb-4a91-bfc2-1510a18499e9.jpg', 'japanese.jpg')
+torch.hub.download_url_to_file('https://img.mghubcdn.com/file/imghub/moriarty-the-patriot/68/10.jpg', 'english.jpg')
     
 def midpoint(x1, y1, x2, y2):
     x_mid = int((x1 + x2)/2)
@@ -21,9 +24,6 @@ def midpoint(x1, y1, x2, y2):
 def draw_mask(img, bounds):
     mask = np.zeros(img.shape[:2], dtype="uint8")
     for bound in bounds:
-        boxes.append(count)
-        count = count + 1
-        box0, box1, box2, box3 = bound[0]
         
         x0, y0 = box0
         x1, y1 = box1 
@@ -37,7 +37,7 @@ def draw_mask(img, bounds):
         
         cv.line(mask, (x_mid0, y_mid0), (x_mid1, y_mi1), 255, thickness)
     img = cv.inpaint(img, mask, 3, cv.INPAINT_TELEA)            
-    return(img, bounds)
+    return(img)
        
 def inference(img, lang):
     if lang == "english":
@@ -60,11 +60,11 @@ def inference(img, lang):
     
     return ['box.jpg', 'result.jpg', pd.DataFrame(bounds). iloc[: , 1:2]]
 
-title = 'EasyOCR'
-description = 'Gradio demo for EasyOCR. EasyOCR demo supports 80+ languages.To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to load them. Read more at the links below.'
-article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/'>Ready-to-use OCR with 80+ supported languages and all popular writing scripts including Latin, Chinese, Arabic, Devanagari, Cyrillic and etc.</a> | <a href='https://github.com/JaidedAI/EasyOCR'>Github Repo</a></p>"
+title = 'Manga Image Cleaner'
+description = 'Image inpainting and text detection demo with the use of EasyOCR and CV2. To use it, simply upload your image and choose a language from the dropdown menu, or click one of the examples to test the program.'
+article = "<p style='text-align: center'><a href='https://www.jaided.ai/easyocr/'>Ready-to-use image inpainting with supported languages such as: Chinese, English, Japanese, and Korean</a> | <a href='https://github.com/JaidedAI/EasyOCR'>Github OCR Repo</a> | <a href='https://towardsdatascience.com/remove-text-from-images-using-cv2-and-keras-ocr-24e7612ae4f4'>CV2 Reference</a></p>"
 css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
-examples = [['korean.jpg',"korean"]]
+examples = [['chinese.jpg',"chinese"], ['english.jpg',"english"], ['japanese.jpg',"japanese"], ['korean.jpg',"korean"]]
 choices = [
     "chinese",
     "english",