muhammadzain commited on
Commit
a8f0294
1 Parent(s): 36da93e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  from flask import Flask, request, send_file, redirect
2
  from werkzeug.utils import secure_filename
3
  from flask_cors import CORS
4
- from rembg import remove
5
  import cv2
6
  import os
7
 
@@ -19,7 +19,7 @@ def inference():
19
  file_name = file.filename
20
  image = cv2.imread(secure_filename(file_name))
21
  w,h = image.shape[1],image.shape[0]
22
- image = cv2.resize(output,(int(w*.79),int(h*.79)))
23
  cv2.imwrite(secure_filename(file.filename),image)
24
  os.system("uname -a")
25
  #output = remove(image) # remove background
 
1
  from flask import Flask, request, send_file, redirect
2
  from werkzeug.utils import secure_filename
3
  from flask_cors import CORS
4
+ #from rembg import remove
5
  import cv2
6
  import os
7
 
 
19
  file_name = file.filename
20
  image = cv2.imread(secure_filename(file_name))
21
  w,h = image.shape[1],image.shape[0]
22
+ image = cv2.resize(image,(int(w*.79),int(h*.79)))
23
  cv2.imwrite(secure_filename(file.filename),image)
24
  os.system("uname -a")
25
  #output = remove(image) # remove background