update dockerfile

Dockerfile

@@ -1,11 +1,11 @@
-# FROM python:3.10 AS builder
-FROM python:3.10
+FROM python:3.10 AS builder
+# FROM python:3.10
 
 WORKDIR /code
 # WORKDIR /home/user/app
 
 # COPY ./requirements.txt /code/requirements.txt
-COPY . /code
+# COPY . /code
 
 RUN apt-get update && apt-get install -y git
 
@@ -21,16 +21,19 @@ RUN git submodule update --recursive
 
 # Stage 2
 
-# FROM python:3.10
+FROM python:3.10
 
 # WORKDIR /home/user/app
-# WORKDIR /code
+WORKDIR /code
 
 # Copy files from the builder stage
 # COPY --from=builder /home/user/app /home/user/app
+COPY --from=builder /code/yachtiocraft /code/yachtiocraft
 
 # COPY ./requirements.txt /home/user/app/requirements.txt
 
+WORKDIR /code/yachtiocraft
+
 RUN pip install torch
 
 RUN pip install wheel