Update Dockerfile

Dockerfile

@@ -10,15 +10,13 @@ COPY . .
 # Install pip and upgrade it
 RUN pip install --upgrade pip
 
-# Install core dependencies including Uvicorn
-RUN pip install packaging torch --extra-index-url https://download.pytorch.org/whl/cu118
-
-# Install additional dependencies including Uvicorn
+# Install core dependencies including FastAPI and Uvicorn
 RUN pip install \
+    fastapi uvicorn \
     nltk sacremoses pandas regex mock \
     "transformers>=4.33.2" mosestokenizer \
     bitsandbytes scipy accelerate datasets \
-    sentencepiece uvicorn
+    sentencepiece
 
 # Download NLTK punkt tokenizer
 RUN python3 -c "import nltk; nltk.download('punkt')"