Update Dockerfile

Dockerfile

@@ -14,7 +14,6 @@ USER user
 
 # Set environmental variables
 ENV PATH="/home/user/.local/bin:$PATH"
-ENV PYTHONUNBUFFERED=1
 
 # Copy in the requirements file and install dependencies
 COPY --chown=user:user requirements.txt .
@@ -22,10 +21,9 @@ RUN pip install --no-cache-dir -r requirements.txt
 
 # Copy the actual Python script and .env if it exists
 COPY --chown=user:user main.py .
-COPY --chown=user:user run.py .
 # We don't forcibly COPY .env because users should use HF Secrets,
 # but if it's there it won't break anything.
 
 # The command to start our infinite loop script
 # The -u flag is REQUIRED for Docker/HuggingFace so logs print immediately!
-CMD ["python", "-u", "run.py"]
+CMD ["python", "-u", "main.py", "--run"]