Update Dockerfile

Dockerfile

@@ -1,40 +1,46 @@
+# Start from the base Python 3.9 image
 FROM python:3.9
 
-FROM nvidia/cuda:12.1.1-cudnn8-devel-ubuntu22.04
-
-ARG DEBIAN_FRONTEND=noninteractive
-
+# Install required packages
 RUN apt-get update && apt-get install --no-install-recommends -y \
-  build-essential \
-  python3.9 \
-  python3-pip \
-  git \
-  ffmpeg \
-  && apt-get clean && rm -rf /var/lib/apt/lists/*
+    build-essential \
+    git \
+    ffmpeg \
+    && apt-get clean && rm -rf /var/lib/apt/lists/*
 
+# Set the working directory
 WORKDIR /code
 
+# Copy the requirements file and install dependencies
 COPY ./requirements.txt /code/requirements.txt
-
-# For hugging face
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
-RUN pip install pyngrok
-
-RUN ngrok config add-authtoken 1XtU01EKWysplGQ8fz54lVUQpnQ_3KKhw6YNQ5E85rxdkgXx4
-
-#set up working directory
+# Add a new user and switch to that user
 RUN useradd -m -u 1000 user
-
 USER user
 
+# Set environment variables
 ENV HOME=/home/user \
-  PATH=/home/user/.local/bin:$PATH
+    PATH=/home/user/.local/bin:$PATH
 
+# Set the working directory
 WORKDIR $HOME/app
 
+# Copy the application files into the container
 COPY --chown=user . $HOME/app
-#end of dir set up
 
+# Install ngrok
+RUN wget https://bin.equinox.io/c/4VmDzA7iaHb/ngrok-stable-linux-amd64.zip && \
+    unzip ngrok-stable-linux-amd64.zip && \
+    mv ngrok $HOME/.local/bin && \
+    rm ngrok-stable-linux-amd64.zip
+
+# Set permissions for ngrok
+RUN chmod +x $HOME/.local/bin/ngrok
+
+# Expose the ports for FastAPI and ngrok's web interface
+EXPOSE 7860
+EXPOSE 4040
 
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--workers", "2"]
+# Run the FastAPI application and ngrok
+CMD ["sh", "-c", "uvicorn app:app --host 0.0.0.0 --port 7860 & sleep 5 && ngrok http 7860"]