Update app.py

app.py

@@ -20,9 +20,10 @@ def inference():
     image = cv2.imread(secure_filename(file_name))
     w,h = image.shape[1],image.shape[0]
     os.system("uname -a")
-    output = remove(image)  # remove background
-    output = cv2.resize(output,(int(w*.79),int(h*.79)))
-    cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.png',output)
+    #output = remove(image)  # remove background
+    os.system('backgroundremover -i'+'{name}'.format(name= secure_filename(file.filename)) + ' -m "u2net_human_seg" -o {name}'.format(name= secure_filename(file.filename)))
+    #output = cv2.resize(output,(int(w*.79),int(h*.79)))
+    #cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.png',output)
     print(secure_filename(file.filename).split('.'))
     os.system("ls -l")
     return send_file(secure_filename(file.filename).split('.')[0]+'.png',mimetype='image/png')