add docker related files

Dockerfile

@@ -0,0 +1,20 @@
+# 使用 Python 官方鏡像 (3.9-slim)
+FROM python:3.9-slim
+
+# 設置工作目錄
+WORKDIR /app
+
+# 安裝系統依賴（解決 opencv-python 問題）
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx libglib2.0-0 && \
+    apt-get clean && rm -rf /var/lib/apt/lists/*
+
+# 複製依賴文件並安裝依賴
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# 複製應用程式代碼到容器
+COPY segmentation.py app.py /app/
+
+# 指定容器啟動時的執行指令
+CMD ["python", "app.py"]

app.py

@@ -1,13 +1,4 @@
-import numpy as np
 import gradio as gr
-import sys,  time
-import ipywidgets as widget
-from IPython.display import display
-import numpy as np
-import cv2
-from PIL import Image as PIL_Image
-from io import BytesIO
-import matplotlib.pyplot as plt
 from segmentation import ImageSegmentation
 
 
@@ -35,4 +26,4 @@ with gr.Blocks() as demo:
     image_button.click(image_segmentation, inputs=[image_input, temp_slider], outputs=image_output)
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(server_name="0.0.0.0", server_port=7860)

requirements.txt

@@ -0,0 +1,4 @@
+gradio
+numpy
+Opencv-python
+pillow

segmentation.py

@@ -1,11 +1,7 @@
-import sys,  time
-import ipywidgets as widget
-from IPython.display import display
 import numpy as np
 import cv2
 from PIL import Image as PIL_Image
 from io import BytesIO
-import matplotlib.pyplot as plt
 
 class ImageSegmentation():
     def __init__(self, x, threshold_factor):