Update Dockerfile

Dockerfile

@@ -11,6 +11,9 @@ RUN apt-get update && apt-get install -y \
     g++ \
     && rm -rf /var/lib/apt/lists/*
 
+# Create symbolic link for python
+RUN ln -s /usr/bin/python3.10 /usr/bin/python
+
 WORKDIR /app
 
 # Install basic Python packages first
@@ -53,8 +56,11 @@ RUN git clone -b self-lengthen https://github.com/quanshr/FastChat.git && \
 # Install LLaMA Factory
 RUN pip3 install --no-cache-dir llamafactory
 
-# Copy project files
-COPY . .
+# Copy project structure
+COPY qwen /app/qwen
+COPY llama /app/llama
+COPY eval /app/eval
+COPY app.py /app/app.py
 
 # Set environment variables
 ENV CUDA_VISIBLE_DEVICES=0
@@ -63,11 +69,18 @@ ENV RANK=0
 ENV MASTER_ADDR=localhost
 ENV MASTER_PORT=29500
 
+# Create directories
+RUN mkdir -p /app/qwen/results
+
 # Create startup script
 RUN echo '#!/bin/bash\n\
-cd /app/qwen\n\
-bash run.sh --base_model=$MODEL_PATH --instruct_count=$INSTRUCT_COUNT --max_iter=$MAX_ITER\n\
-python collect_data.py' > /app/start.sh && \
+if [ "$USE_UI" = "true" ]; then\n\
+    python app.py\n\
+else\n\
+    cd /app/qwen\n\
+    bash run.sh --base_model=$MODEL_PATH --instruct_count=$INSTRUCT_COUNT --max_iter=$MAX_ITER\n\
+    python collect_data.py\n\
+fi' > /app/start.sh && \
 chmod +x /app/start.sh
 
 # Command to run