Rename RemoveText.py to ap.py.py

RemoveText.py → ap.py.py

@@ -1,12 +1,10 @@
 
-
 from cv2 import threshold
-import matplotlib.pyplot as plt
 import keras_ocr
 import cv2
 import math
 import numpy as np
-
+import gradio as gr
 
 def midpoint(x1, y1, x2, y2):
     x_mid = int((x1 + x2)/2)
@@ -62,9 +60,14 @@ def inpaint_text(img_path, pipeline):
 # keras-ocr will automatically download pretrained
 # weights for the detector and recognizer.
 pipeline = keras_ocr.pipeline.Pipeline()
+def RemoveText(image):
+    img_text_removed = inpaint_text(image, pipeline)
+    return  cv2.cvtColor(img_text_removed, cv2.COLOR_BGR2RGB)
 
-img_text_removed = inpaint_text('1.jpg', pipeline)
-
-plt.imshow(img_text_removed)
+iface = gr.Interface(fn=ApplyChange,
+             inputs=gr.inputs.Image(label="Image to Remove Text From", type="numpy"),
+             outputs="image",
+                    examples=[["1.jpg"]],
+                    title="Remove Text for Image")
+iface.launch(debug=True)
 
-cv2.imwrite('text_removed_image.jpg', cv2.cvtColor(img_text_removed, cv2.COLOR_BGR2RGB))