Update Dockerfile

Dockerfile

@@ -1,10 +1,12 @@
 FROM python:3.9
 
 # Create app directory
-WORKDIR /code
+WORKDIR /app
 
-RUN wget https://huggingface.co/spaces/anon5/vit-base-patch32-384-api/raw/main/app.py -O ./app.py
-RUN pip3 install flask transformers
+COPY ./requirements.txt /app
+COPY ./app.py /app
+
+RUN pip install -r requirements.txt
 
 EXPOSE 5000
-CMD [ "python", "-m", "flask ", "--app", "app", "run" ]
+CMD ["flask", "run", "--host", "0.0.0.0"]