Update app.py

app.py

@@ -34,9 +34,9 @@ def mainTest(bgra, outpath):
 
 def deep_nude_process(item):
     # print('Processing {}'.format(item))
-    #dress = cv2.imread(item)
+    dress = cv2.imread(item)
     #dress = cv2.cvtColor(dress, cv2.COLOR_BGRA2RGBA)
-    dress = item
+    #dress = item
     h = dress.shape[0]
     w = dress.shape[1]
     dress = cv2.resize(dress, (512, 512), interpolation=cv2.INTER_CUBIC)
@@ -48,13 +48,13 @@ def deep_nude_process(item):
 def inference(img):
     global index
     bgra = cv2.cvtColor(img, cv2.COLOR_RGBA2BGRA)
-    #inputpath = "input_" + str(index) + ".jpg"
-    #cv2.imwrite(inputpath, bgra)
+    inputpath = "input_" + str(index) + ".jpg"
+    cv2.imwrite(inputpath, bgra)
 
     outputpath = "out_" + str(index) + ".jpg"
     index += 1
     print(time.strftime("开始!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
-    output = mainTest(bgra, outputpath)
+    output = mainTest(inputpath, outputpath)
     print(time.strftime("结束!!!!!!!!! %Y-%m-%d %H:%M:%S", time.localtime()))
     return output