Update Dockerfile

Dockerfile

@@ -1,41 +1,3 @@
-# FROM nvidia/cuda:12.1.0-base-ubuntu22.04
-
-# # Install system dependencies
-# RUN apt-get update && apt-get install -y \
-#     python3.10 \
-#     python3-pip \
-#     git \
-#     && rm -rf /var/lib/apt/lists/*
-
-# WORKDIR /app
-
-# # Create cache directory
-# RUN mkdir -p /.cache/huggingface && \
-#     chmod -R 777 /.cache
-
-# # Create Streamlit config directory
-# RUN mkdir -p /app/.streamlit
-
-# # Install chandra-ocr
-# RUN pip3 install --no-cache-dir chandra-ocr
-
-# # Expose port
-# EXPOSE 7860
-
-# # Environment variables
-# ENV STREAMLIT_SERVER_PORT=7860
-# ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
-# ENV STREAMLIT_SERVER_HEADLESS=true
-# ENV HF_HOME=/.cache/huggingface
-# ENV TRANSFORMERS_CACHE=/.cache/huggingface
-# ENV HF_DATASETS_CACHE=/.cache/huggingface
-
-# # Copy Streamlit config (will be available from the repo)
-# COPY .streamlit/config.toml /app/.streamlit/config.toml
-
-# # Run the app
-# CMD ["python3", "-m", "streamlit", "run", "/usr/local/lib/python3.10/dist-packages/chandra/scripts/app.py", "--server.port=7860", "--server.address=0.0.0.0"]
-
 FROM nvidia/cuda:12.1.0-base-ubuntu22.04
 
 # Install system dependencies
@@ -60,27 +22,17 @@ RUN pip3 install --no-cache-dir chandra-ocr
 # Expose port
 EXPOSE 7860
 
-# Environment variables for Streamlit
+# Environment variables
 ENV STREAMLIT_SERVER_PORT=7860
 ENV STREAMLIT_SERVER_ADDRESS=0.0.0.0
 ENV STREAMLIT_SERVER_HEADLESS=true
-
-# HuggingFace cache
 ENV HF_HOME=/.cache/huggingface
 ENV TRANSFORMERS_CACHE=/.cache/huggingface
 ENV HF_DATASETS_CACHE=/.cache/huggingface
 
-# CRITICAL: Memory optimization settings
-ENV PYTORCH_CUDA_ALLOC_CONF=expandable_segments:True
-ENV CUDA_LAUNCH_BLOCKING=0
-
-# Load model in 8-bit precision to save memory
-ENV MODEL_PRECISION=8bit
-
+# Copy Streamlit config (will be available from the repo)
 COPY .streamlit/config.toml /app/.streamlit/config.toml
 
-
 # Run the app
 CMD ["python3", "-m", "streamlit", "run", "/usr/local/lib/python3.10/dist-packages/chandra/scripts/app.py", "--server.port=7860", "--server.address=0.0.0.0"]
 
-# CMD ["python3", "-m", "streamlit", "run", "/usr/local/lib/python3.10/dist-packages/chandra/scripts/app.py", "--server.port=7860", "--server.address=0.0.0.0", "--server.enableXsrfProtection=false", "--server.enableCORS=false"]