Updated to a smaller container

.dockerignore

@@ -0,0 +1,2 @@
+/venv
+/flagged

Dockerfile

@@ -1,26 +1,18 @@
-FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04
+FROM continuumio/miniconda3
 EXPOSE 7860
 
-ENV CONDA_DIR /opt/conda
 RUN apt-get update && \
-    apt-get install -y build-essential  && \
-    apt-get install -y wget && \
-    apt-get install -y libgl1-mesa-glx && \
-    apt-get install -y libglib2.0-0 && \
+    apt-get install -y --no-install-recommends libgl1-mesa-glx && \
+    apt-get install -y --no-install-recommends libglib2.0-0 && \
     apt-get clean && \
     rm -rf /var/lib/apt/lists/*
 
-RUN wget --quiet https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda.sh && \
-     /bin/bash ~/miniconda.sh -b -p /opt/conda
-
-ENV PATH=$CONDA_DIR/bin:$PATH
-
 ENV GRADIO_SERVER_NAME=0.0.0.0
 WORKDIR /workspace
 
 ADD environment.yaml /workspace/environment.yaml
-RUN conda env update -n base --file environment.yaml
-RUN pip install typing-extensions -U
+RUN conda env update -n base --file environment.yaml && \
+    pip install typing-extensions -U
 
 ADD app.py /workspace/
 CMD [ "python" , "/workspace/app.py" ]

app.py

@@ -1,6 +1,5 @@
 from fawkes.protection import Fawkes
 import gradio as gr
-import os
 
 def predict(level, img):
   # print(img)
@@ -10,4 +9,6 @@ def predict(level, img):
   # print(os.listdir('/tmp'))
   return splt[0] + "_cloaked." + splt[1]
 
-gr.Interface(fn=predict, inputs=[gr.inputs.Dropdown(["low", "mid", "high"], label="Protection Level"), gr.inputs.Image(type='filepath')], outputs=gr.outputs.Image(type="pil")).launch()
+gr.Interface(fn=predict, inputs=[gr.components.Dropdown(["low", "mid", "high"], label="Protection Level"),
+                                 gr.components.Image(type='filepath')],
+                                 outputs=gr.components.Image(type="pil")).launch(show_error=True)

requirements.txt

@@ -1 +1,95 @@
-fawkes==1.0.4
+absl-py==0.15.0
+aiohttp==3.8.3
+aiosignal==1.3.1
+analytics-python==1.4.0
+anyio==3.6.2
+astunparse==1.6.3
+async-timeout==4.0.2
+asynctest==0.13.0
+attrs==22.1.0
+backoff==1.10.0
+bcrypt==4.0.1
+bleach==5.0.1
+cachetools==5.2.0
+certifi==2022.9.24
+cffi==1.15.1
+charset-normalizer==2.1.1
+click==8.1.3
+cryptography==38.0.3
+cycler==0.11.0
+fastapi==0.76.0
+fawkes==1.0.4
+ffmpy==0.3.0
+flatbuffers==1.12
+fonttools==4.38.0
+frozenlist==1.3.3
+fsspec==2022.10.0
+gast==0.3.3
+google-auth==2.14.1
+google-auth-oauthlib==0.4.6
+google-pasta==0.2.0
+gradio==3.0.20
+grpcio==1.32.0
+h11==0.14.0
+h5py==2.10.0
+idna==3.4
+importlib-metadata==5.0.0
+Jinja2==3.1.2
+Keras==2.4.3
+Keras-Preprocessing==1.1.2
+kiwisolver==1.4.4
+linkify-it-py==1.0.3
+Markdown==3.4.1
+markdown-it-py==2.1.0
+MarkupSafe==2.1.1
+matplotlib==3.5.3
+mdit-py-plugins==0.3.1
+mdurl==0.1.2
+monotonic==1.6
+mtcnn==0.1.1
+multidict==6.0.2
+numpy==1.19.5
+oauthlib==3.2.2
+opencv-python==4.6.0.66
+opt-einsum==3.3.0
+orjson==3.8.1
+packaging==21.3
+pandas==1.3.5
+paramiko==2.12.0
+Pillow==9.3.0
+pkg_resources==0.0.0
+protobuf==3.20.3
+pyasn1==0.4.8
+pyasn1-modules==0.2.8
+pycparser==2.21
+pycryptodome==3.15.0
+pydantic==1.9.2
+pydub==0.25.1
+PyNaCl==1.5.0
+pyparsing==3.0.9
+python-dateutil==2.8.2
+python-multipart==0.0.5
+pytz==2022.6
+PyYAML==6.0
+requests==2.28.1
+requests-oauthlib==1.3.1
+rsa==4.9
+scipy==1.7.3
+six==1.15.0
+sniffio==1.3.0
+starlette==0.18.0
+tensorboard==2.11.0
+tensorboard-data-server==0.6.1
+tensorboard-plugin-wit==1.8.1
+tensorflow==2.4.1
+tensorflow-estimator==2.4.0
+termcolor==1.1.0
+typing_extensions==4.4.0
+uc-micro-py==1.0.1
+urllib3==1.26.12
+uvicorn==0.19.0
+webencodings==0.5.1
+Werkzeug==2.2.2
+wrapt==1.12.1
+yarl==1.8.1
+zipp==3.10.0