Update app.py

app.py

@@ -12,6 +12,8 @@ from urllib.request import urlretrieve
 from io import BytesIO
 import numpy as np
 import torchvision.transforms as T
+from PIL import Image,ImageOps,ImageFilter
+
 
 
 
@@ -58,4 +60,12 @@ if uploaded_file is not None:
   img = PIL.Image.fromarray(x).convert('RGB')
   size = a.size
   im1 = img.resize(size)
-  st.image(im1, caption='PoP ArT')
+  im1.save("test.png")
+  im = Image.open('/content/test.png')
+  im = im.convert('RGBA')
+  data = np.array(im)   # "data" is a height x width x 4 numpy array
+  red, green, blue, alpha = data.T # Temporarily unpack the bands for readability'
+  white_areas = (red == 0) & (blue == 0) & (green == 0)
+  data[..., :-1][white_areas.T] = (105,0,0) # Transpose back needed
+  im2 = Image.fromarray(data)
+  st.image(im2, caption='PoP ArT')