Create Dockerfile

Dockerfile

@@ -0,0 +1,36 @@
+# Use the official Python 3.9 image
+FROM python:3.9
+
+# Set the working directory
+WORKDIR /app
+
+# Copy your FastAPI code and setup scripts
+COPY . .
+
+# Install dependencies via pip
+RUN pip install --upgrade pip
+
+# Install PyTorch with CUDA support
+RUN pip install torch --extra-index-url https://download.pytorch.org/whl/cu118
+
+# Install other dependencies
+RUN pip install \
+    nltk sacremoses pandas regex mock \
+    "transformers>=4.33.2" mosestokenizer \
+    bitsandbytes scipy accelerate datasets \
+    "flash-attn>=2.1" sentencepiece
+
+# Download NLTK punkt tokenizer
+RUN python3 -c "import nltk; nltk.download('punkt')"
+
+# Clone and 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 7860
+
+# Start the FastAPI app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]