muhammadzain commited on
Commit
ad8d51e
1 Parent(s): fa1eec9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -3,6 +3,7 @@ from werkzeug.utils import secure_filename
3
  from flask_cors import CORS
4
  from rembg import remove
5
  import cv2
 
6
 
7
  app = Flask(__name__)
8
  CORS(app)
@@ -20,6 +21,7 @@ def inference():
20
  output = remove(image) # remove background
21
  cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.png',output)
22
  print(secure_filename(file.filename).split('.'))
 
23
  return send_file(secure_filename(file.filename).split('.')[0]+'.png',mimetype='image/png')
24
 
25
 
 
3
  from flask_cors import CORS
4
  from rembg import remove
5
  import cv2
6
+ import os
7
 
8
  app = Flask(__name__)
9
  CORS(app)
 
21
  output = remove(image) # remove background
22
  cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.png',output)
23
  print(secure_filename(file.filename).split('.'))
24
+ os.system("ls -l")
25
  return send_file(secure_filename(file.filename).split('.')[0]+'.png',mimetype='image/png')
26
 
27