Update Dockerfile

Dockerfile

@@ -4,36 +4,33 @@ FROM python:3.9
 # Set the working directory
 WORKDIR /app
 
-# Copy your FastAPI code and setup scripts
+# Copy code and scripts
 COPY . .
 
-# Install pip and upgrade it
+# Install pip and upgrade
 RUN pip install --upgrade pip
 
-# Install the 'packaging' library first
-RUN pip install packaging
+# Install core dependencies
+RUN pip install packaging torch --extra-index-url https://download.pytorch.org/whl/cu118
 
-# Install PyTorch with CUDA support
-RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu118
-
-# Install all other dependencies (re-ordered)
+# Install additional dependencies without `flash-attn`
 RUN pip install \
     nltk sacremoses pandas regex mock \
     "transformers>=4.33.2" mosestokenizer \
     bitsandbytes scipy accelerate datasets \
-    sentencepiece "flash-attn>=2.1"
+    sentencepiece
 
 # Download NLTK punkt tokenizer
 RUN python3 -c "import nltk; nltk.download('punkt')"
 
-# Clone and install IndicTrans2 toolkit
+# Install IndicTrans2 toolkit
 RUN git clone https://github.com/VarunGumma/IndicTransToolkit && \
     cd IndicTransToolkit && \
     pip install --editable ./ && \
     cd ..
 
-# Expose port 7860 for FastAPI (or change if needed)
+# Expose port 7860
 EXPOSE 7860
 
-# Start the FastAPI app
+# Start FastAPI server
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]