backend 0

Dockerfile

@@ -0,0 +1,24 @@
+FROM python:3.10
+
+RUN apt-get update -y && apt-get install -y build-essential
+
+WORKDIR /app
+
+RUN useradd -m -u 1000 user
+USER user
+ENV HOME=/home/user \
+	PATH=/home/user/.local/bin:$PATH
+
+WORKDIR $HOME/app
+
+COPY --chown=user . $HOME/app
+
+COPY app.py app.py
+
+RUN pip install Flask
+RUN pip install gunicorn
+RUN pip install -U flask-cors
+RUN pip install opencv-python-headless==4.5.5.64
+RUN pip install rembg
+
+CMD ["gunicorn","-b","0.0.0.0:7860", "app:app","--timeout","950"]

app.py

@@ -0,0 +1,27 @@
+from flask import Flask, request, send_file
+from werkzeug.utils import secure_filename
+from flask_cors import CORS
+from rembg import remove
+import cv2
+
+app = Flask(__name__)
+CORS(app)
+
+@app.route('/')
+def index():
+    return "Please use frontend"
+
+@app.route('/data',methods = ['POST'])
+def inference():
+    file = request.files['file']
+    file.save(secure_filename(file.filename))
+    file_name = file.filename
+    image = cv2.imread(secure_filename(file_name))
+    output = remove(image)  # remove background
+    cv2.imwrite(secure_filename(file.filename).split('.')[0]+'.png',output)
+    print(secure_filename(file.filename).split('.'))
+    return send_file(secure_filename(file.filename).split('.')[0]+'.png',mimetype='image/png')
+
+
+if __name__ == "__main__":
+    app.run(debug=True,host="0.0.0.0",port=5000)