Update Dockerfile

Dockerfile

@@ -1,14 +1,10 @@
-# 1. Use a base image with Python and CUDA (for the GPU)
 FROM python:3.11-slim
 
-# 2. Set the working directory
 WORKDIR /code
-
-# 3. Copy all your application files into the container
 COPY . .
 
-# 4. Install the dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 5. Run the fine-tuning script when the Space starts
-CMD ["python", "fine_tune.py"]
+EXPOSE 7860
+
+CMD ["python", "fine_tune.py"]