Upload with huggingface_hub

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.8-slim-buster
+
+WORKDIR /app
+
+COPY requirements.txt .
+
+RUN pip3 install --no-cache-dir -r requirements.txt
+
+COPY . .
+
+CMD ["python3", "app.py"]

app.py

@@ -0,0 +1,9 @@
+import gradio as gr
+import cv2
+
+def color_to_bw(image):
+    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    return gray
+
+iface = gr.Interface(fn=color_to_bw, inputs="image", outputs="image")
+iface.launch()

requirements.txt

@@ -0,0 +1,2 @@
+gradio
+opencv-python-headless

test.py

@@ -0,0 +1,10 @@
+import cv2
+import numpy as np
+from app import color_to_bw
+
+def test_color_to_bw():
+    image = cv2.imread("test_image.jpg")
+    expected_output = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
+    assert np.array_equal(color_to_bw(image), expected_output)
+
+test_color_to_bw()