Create Dockerfile

Dockerfile

@@ -0,0 +1,12 @@
+# Use a PyTorch base image
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
+
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --upgrade pip && pip install -r requirements.txt
+
+COPY app.py .
+
+EXPOSE 7860
+CMD ["python", "app.py"]