Upload Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# Use an official Python runtime as a parent image
-FROM python:3.10-slim
+# Use Python 3.9 as base image
+FROM python:3.9
 
 # Set working directory within the container
 WORKDIR /code
@@ -10,7 +10,7 @@ COPY ./requirements.txt /code/requirements.txt
 # Install Python dependencies
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-# Install ffmpeg
+# Install ffmpeg (if needed)
 RUN apt update && apt install -y ffmpeg
 
 # Create a non-root user
@@ -32,5 +32,8 @@ COPY --chown=user . $HOME/app
 # Expose port 7860 (assuming your FastAPI app runs on this port)
 EXPOSE 7860
 
+# Ensure PyTorch is installed
+RUN pip install torch torchvision torchaudio
+
 # Command to run the FastAPI application
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]