Update Dockerfile

Dockerfile

@@ -1,17 +1,17 @@
-# Use Python as base image
+# Use an official Python runtime as the base image
 FROM python:3.10
 
-# Set working directory
+# Set the working directory in the container
 WORKDIR /app
 
-# Copy project files
-COPY . /app
+# Copy the FastAPI application code
+COPY server.py /app/server.py
 
-# Install dependencies
-RUN pip install --no-cache-dir fastapi uvicorn torch torchvision torchaudio transformers sentencepiece accelerate
+# Install necessary dependencies
+RUN pip install --no-cache-dir fastapi uvicorn transformers torch sentencepiece
 
-# Expose port 7860 for FastAPI
+# Expose the FastAPI port
 EXPOSE 7860
 
-# Start the FastAPI server
+# Run the FastAPI server
 CMD ["uvicorn", "server:app", "--host", "0.0.0.0", "--port", "7860"]