Update app.py

app.py

@@ -13,6 +13,7 @@ from io import BytesIO
 import numpy as np
 import torchvision.transforms as T
 from PIL import Image,ImageOps,ImageFilter
+from io import BytesIO
 
 
 
@@ -60,8 +61,9 @@ if uploaded_file is not None:
   img = PIL.Image.fromarray(x).convert('RGB')
   size = a.size
   im1 = img.resize(size)
-  im1.save("test.png")
-  im = Image.open('/content/test.png')
+  membuf = BytesIO()
+  im1.save(membuf, format="png") 
+  im = Image.open(membuf)
   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'