Update app.py

app.py

@@ -24,19 +24,16 @@ main.py
 """
 
 
-def mainTest(bgra, outpath):
-    watermark = deep_nude_process(bgra)
-    #watermark1 = cv2.cvtColor(watermark, cv2.COLOR_RGBA2BGRA)
+def mainTest(inputpath, outpath):
+    watermark = deep_nude_process(inputpath)
+    watermark1 = cv2.cvtColor(watermark, cv2.COLOR_BGRA2RGBA)
     #cv2.imwrite(outpath, watermark1)
-    return watermark
+    return watermark1
     #
 
 
-def deep_nude_process(item):
-    # print('Processing {}'.format(item))
-    dress = cv2.imread(item)
-    #dress = cv2.cvtColor(dress, cv2.COLOR_BGRA2RGBA)
-    #dress = item
+def deep_nude_process(inputpath):
+    dress = cv2.imread(inputpath)
     h = dress.shape[0]
     w = dress.shape[1]
     dress = cv2.resize(dress, (512, 512), interpolation=cv2.INTER_CUBIC)