NeuralFalcon commited on
Commit
104be4a
·
verified ·
1 Parent(s): 89a587e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -55,7 +55,9 @@ def remove_watermark(image_path,file_type="",saved_path=""):
55
  # cv2.imwrite('watermark_area.jpg', watermark_area)
56
 
57
  # Create the mask for the watermark area
58
- text_mask_path = 'watermark_mask.png'
 
 
59
  cleaned_image = remove_watermark_area(watermark_area, text_mask_path)
60
  # cv2.imwrite('cleaned_watermark.jpg', cleaned_image)
61
  # Paste back the cleaned watermark on the original image
 
55
  # cv2.imwrite('watermark_area.jpg', watermark_area)
56
 
57
  # Create the mask for the watermark area
58
+ # text_mask_path = 'watermark_mask.png'
59
+ # text_mask_path ='./mask/mask_1.png'
60
+ text_mask_path ='./mask/mask_2.png'
61
  cleaned_image = remove_watermark_area(watermark_area, text_mask_path)
62
  # cv2.imwrite('cleaned_watermark.jpg', cleaned_image)
63
  # Paste back the cleaned watermark on the original image