Update Dockerfile

Dockerfile

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