Update Dockerfile

Dockerfile

@@ -1,14 +1,13 @@
-FROM python:3.9.0
+from condaforge/mambaforge
 ENV MPLCONFIGDIR=".config/matplotlib"
 RUN mkdir -m 777 .cache .config ./flagged
 RUN git clone https://github.com/szymonrucinski/AiRoll
 WORKDIR /AiRoll
-RUN git checkout update-reqs
 RUN git clone https://huggingface.co/szymonrucinski/what-a-shot
 RUN ls
 RUN mv ./what-a-shot ./model
-RUN pip install -r requirements.txt
+RUN conda env create -f environment.yaml && conda info
+RUN mamba init bash
 RUN mkdir -m 777 flagged
-RUN ls
 EXPOSE 7860
-CMD ["gradio","start_api.py"]
+CMD ["mamba", "run", "-n", "airoll", "gradio", "start_api.py"]