Update Dockerfile

Dockerfile

@@ -1,8 +1,20 @@
-FROM ghcr.io/replicate/cog:0.5.0
+FROM python:3.9-slim
 
-COPY . /src
+# Install system packages
+RUN apt-get update && apt-get install -y ffmpeg imagemagick
+
+# Install pip packages
+RUN pip install cog
+
+# Set the working directory
 WORKDIR /src
 
+# Copy the requirements.txt file and install Python packages
+COPY requirements.txt .
 RUN pip install -r requirements.txt
 
-ENTRYPOINT ["python", "predict.py"]
+# Copy the rest of the application code
+COPY . .
+
+# Set the entry point for Cog
+ENTRYPOINT ["cog", "predict"]