init

.env

@@ -0,0 +1,12 @@
+# Environment variables for docker-compose.yml
+LOG_LEVEL=DEBUG
+
+## Host
+HOST=localhost
+APP_ENV=prod
+
+# App Config
+MK_APP_ID=hf-background-remover
+APP_PATH=/
+APP_HTTP_PORT=1070
+APP_HTTPS_PORT=5070

Dockerfile

@@ -0,0 +1,20 @@
+# start with a base image
+FROM python:3.8.9
+
+RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6 cmake libgl1-mesa-glx
+WORKDIR /home/user/app
+RUN mkdir app
+RUN pip install pip==22.0.2
+WORKDIR /home/user
+RUN apt-get update && apt-get install -y git-lfs
+COPY packages.txt /root/packages.txt
+COPY requirements.txt /home/user/app/requirements.txt
+RUN git lfs install
+RUN apt-get update && xargs -r -a /root/packages.txt apt-get install -y
+COPY ./ /home/user/app
+RUN pip install -r /home/user/app/requirements.txt
+# COPY --from=lfs /app /home/user/app
+WORKDIR /home/user/app
+
+EXPOSE 7860
+CMD ["python", "app.py"]    

README.md

@@ -1,13 +1,13 @@
 ---
-title: Hf Background Remover
-emoji: 📉
-colorFrom: gray
-colorTo: green
-sdk: streamlit
-sdk_version: 1.15.2
+title: rembg
+emoji: 🖼️✂️
+colorFrom: purple
+colorTo: red
+sdk: gradio
+sdk_version: 2.9.4
 app_file: app.py
 pinned: false
-license: cc
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# rembg
+

app.py

@@ -0,0 +1,53 @@
+
+from flask import Flask, jsonify, request
+import cv2
+from rembg.session_factory import new_session
+from rembg import remove
+import requests
+import base64
+import numpy as np
+import streamlit as st
+
+app = Flask(__name__)
+
+@app.route("/")
+def indexApi():
+    return jsonify({"output": "okay"})
+
+@app.route("/remove")
+def removeApi():
+    image_url = request.form.get("image_url")
+    key = request.form.get("key")
+    model = request.form.get("model")
+
+    if (model == None):
+        model = "u2netp"
+
+    # reject if key not the same 
+    apiKey = st.secrets["Api-Key"]
+    if apiKey != key:
+        return jsonify({
+            "image_url": image_url,
+            "model": model,
+            "result": "Invalid API Key",
+        }), 400
+
+
+    response = requests.get(image_url)
+
+    arr = np.asarray(bytearray(response.content), dtype=np.uint8)
+    image = cv2.imdecode(arr, -1) # 'Load it as it is'
+
+    result = remove(image, session=new_session(model))
+
+    #encode result 
+    base64_string = base64.b64encode(cv2.imencode('.png', result)[1]).decode()
+
+    return jsonify({
+        "image_url": image_url,
+        "model": model,
+        "result": base64_string,
+    })
+
+if __name__ == "__main__":
+    app.run(host="0.0.0.0", port=7860)

docker-compose.yml

@@ -0,0 +1,24 @@
+version: "3.9"
+services:
+  app:
+    build: . # uncomment if you're building image using Dockerfile
+    # image: huedaya/flask-opencv:latest
+    container_name: app-${MK_APP_ID}
+    volumes:
+      - ./:/home/user/app/
+    ports:
+      - 7860:7860
+      - 7861:7861
+    networks:
+      - proxy
+    labels:
+      - "traefik.http.routers.${MK_APP_ID}-secure.rule=(Host(`${HOST}`))"
+      - "traefik.http.routers.${MK_APP_ID}-secure.entrypoints=websecure"
+      - "traefik.docker.network=proxy"
+      - "traefik.enable=true"
+      - "traefik.http.middlewares.${MK_APP_ID}-ratelimit.ratelimit.average=5"
+      - "traefik.http.middlewares.${MK_APP_ID}-ratelimit.ratelimit.burst=2"
+
+networks:
+  proxy:
+    external: true

packages.txt

@@ -0,0 +1,3 @@
+ffmpeg 
+libsm6 
+libxext6

requirements.txt

@@ -0,0 +1,5 @@
+requests
+streamlit
+opencv-python
+rembg
+flask