Upload 2 files

Dockerfile

@@ -0,0 +1,13 @@
+# Dockerfile
+FROM python:3.9-slim
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . /code/
+
+# Use gunicorn to run the app. It will look for the 'app' instance in the 'app.py' file.
+# We bind it to 0.0.0.0:7860, which is the port Hugging Face Spaces expects.
+CMD ["gunicorn", "--bind", "0.0.0.0:7860", "app:app"]

requirements.txt

@@ -0,0 +1,5 @@
+flask
+torch
+torchvision
+Pillow
+gunicorn