Create Dockerfile

Dockerfile

@@ -0,0 +1,20 @@
+FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
+
+# Install system dependencies
+RUN apt-get update && apt-get install -y \
+    python3.10 python3-pip git ffmpeg \
+    && rm -rf /var/lib/apt/lists/*
+
+WORKDIR /app
+
+# Copy requirements and install
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Force clean onnxruntime-gpu install
+RUN pip uninstall -y onnxruntime onnxruntime-gpu && \
+    pip install onnxruntime-gpu
+
+COPY . .
+
+CMD ["python3", "app.py"]