Update Dockerfile

Dockerfile

@@ -1,13 +1,14 @@
 FROM python:3.9-slim
 
 # Install system packages
-RUN apt-get update && apt-get install -y ffmpeg imagemagick
+RUN apt-get update && apt-get install -y \
+    ffmpeg \
+    imagemagick \
+    && apt-get clean \
+    && rm -rf /var/lib/apt/lists/*
 
 # Install pip packages
-RUN pip install cog gradio==3.50.2
-
-# Ensure cog is in the PATH
-RUN ln -s /usr/local/bin/cog /usr/bin/cog
+RUN pip install gradio==3.50.2
 
 # Set the working directory
 WORKDIR /src
@@ -15,5 +16,5 @@ WORKDIR /src
 # Copy the rest of the application code
 COPY . /src
 
-# Set the entry point for Cog
-ENTRYPOINT ["cog", "predict"]
+# Set the entry point to run the prediction script
+ENTRYPOINT ["python", "predict.py"]